Commit c5146bb2 authored by okiss's avatar okiss
Browse files

Update utils.py

parent 67a83738
......@@ -41,7 +41,7 @@ def plot_loss(hist,name,val=False):
# haven't write about the pytorch=True part
def save_logs(hist, output_directory=config['model_dir'], pytorch=False):
if pytorch:
# Maybe Oriel can work on this
pass
else:
hist_df = pd.DataFrame(hist.history)
......@@ -70,4 +70,4 @@ def save_model_param(output_directory=config['model_dir'], pytorch=False):
torch.save(net.state_dict(), output_directory + config['model'] + '_model.pth')
else:
classifier.save(output_directory + config['model'] + '_model.h5')
\ No newline at end of file
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment