Skip to content

Commit

Permalink
Remove save function
Browse files Browse the repository at this point in the history
  • Loading branch information
enmanuelmag authored Nov 27, 2022
1 parent d6718a1 commit c660ede
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions iaflow/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -285,7 +285,7 @@ def add_model(
}

self.models[model_name] = model_data
self.save()
#self.__save()
print(f'Model {model_name}/{run_id} was added')
return model_data[run_id]

Expand Down Expand Up @@ -339,7 +339,7 @@ def delete_model(self, model_data, delete_folder: bool = False):
if len(model) == 0:
del self.models[model_name]

self.save()
#self.__save()
print(f'Model {model_name}/{run_id} was deleted')

def train(
Expand Down Expand Up @@ -495,4 +495,4 @@ def plot_history(self, history, path, monitor='val_loss'):
fig.tight_layout()
plt.show()
if path is not None:
fig.savefig(path)
fig.savefig(path)

0 comments on commit c660ede

Please sign in to comment.