Commit f9db7c36 authored by Martin Lebourdais's avatar Martin Lebourdais
Browse files

Toujours des bugs

parent c6d986f7
...@@ -416,7 +416,7 @@ def seqTrain(dataset_yaml, ...@@ -416,7 +416,7 @@ def seqTrain(dataset_yaml,
'epoch': epoch, 'epoch': epoch,
'model_state_dict': model.state_dict(), 'model_state_dict': model.state_dict(),
'optimizer_state_dict': optimizer.state_dict(), 'optimizer_state_dict': optimizer.state_dict(),
'accuracy': best_accuracy, 'accuracy': best_fmes,
'scheduler': scheduler 'scheduler': scheduler
}, is_best, filename=tmp_model_name + ".pt", best_filename=best_model_name + '.pt') }, is_best, filename=tmp_model_name + ".pt", best_filename=best_model_name + '.pt')
else: else:
...@@ -424,7 +424,7 @@ def seqTrain(dataset_yaml, ...@@ -424,7 +424,7 @@ def seqTrain(dataset_yaml,
'epoch': epoch, 'epoch': epoch,
'model_state_dict': model.module.state_dict(), 'model_state_dict': model.module.state_dict(),
'optimizer_state_dict': optimizer.state_dict(), 'optimizer_state_dict': optimizer.state_dict(),
'accuracy': best_accuracy, 'accuracy': best_fmes,
'scheduler': scheduler 'scheduler': scheduler
}, is_best, filename=tmp_model_name + ".pt", best_filename=best_model_name + '.pt') }, is_best, filename=tmp_model_name + ".pt", best_filename=best_model_name + '.pt')
......
Markdown is supported
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