fix: save latest model every epoch

pull/1602/head
gaotingquan 2021-12-27 06:30:01 +00:00 committed by Tingquan Gao
parent d69a6e8f24
commit 7da2a997e9
1 changed files with 8 additions and 8 deletions

View File

@ -312,14 +312,14 @@ class Engine(object):
self.output_dir,
model_name=self.config["Arch"]["name"],
prefix="epoch_{}".format(epoch_id))
# save the latest model
save_load.save_model(
self.model,
self.optimizer, {"metric": acc,
"epoch": epoch_id},
self.output_dir,
model_name=self.config["Arch"]["name"],
prefix="latest")
# save the latest model
save_load.save_model(
self.model,
self.optimizer, {"metric": acc,
"epoch": epoch_id},
self.output_dir,
model_name=self.config["Arch"]["name"],
prefix="latest")
if self.vdl_writer is not None:
self.vdl_writer.close()