Skip to content

O
OvoTools
Commit f350b375 authored by Ilya Ovodov's avatar Ilya Ovodov
Browse files
Options

always save last model

parent 638533e0
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
ovotools/ignite_tools.py
View file @ f350b375
...@@ -146,6 +146,7 @@ class LogTrainingResults: ...@@ -146,6 +146,7 @@ class LogTrainingResults:
engine.state.metrics[key+':'+k] = v engine.state.metrics[key+':'+k] = v
if self.best_model_buffer: if self.best_model_buffer:
self.best_model_buffer(engine) self.best_model_buffer(engine)
self.best_model_buffer.save_model(filename="last.t7")
if event == Events.ITERATION_COMPLETED: if event == Events.ITERATION_COMPLETED:
str = "Epoch:{}.{}\t".format(engine.state.epoch, engine.state.iteration) str = "Epoch:{}.{}\t".format(engine.state.epoch, engine.state.iteration)
else: else:
......
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