diff --git a/crabnet/crabnet_.py b/crabnet/crabnet_.py index 5529707..aa99e24 100644 --- a/crabnet/crabnet_.py +++ b/crabnet/crabnet_.py @@ -1094,8 +1094,8 @@ def save_network(self, model_name: str = None): """ if model_name is None: model_name = self.model_name - os.makedirs(join("models", "trained_models"), exist_ok=True) - path = join("models", "trained_models", f"{model_name}.pth") + os.makedirs(join("models", "trained_models"), exist_ok=True) + path = join("models", "trained_models", f"{model_name}.pth") if self.verbose: print(f"Saving network ({model_name}) to {path}")