|
|
@ -162,7 +162,6 @@ def train_sp(args, config):
|
|
|
|
trainer.extend(VisualDL(output_dir), trigger=(1, "iteration"))
|
|
|
|
trainer.extend(VisualDL(output_dir), trigger=(1, "iteration"))
|
|
|
|
trainer.extend(
|
|
|
|
trainer.extend(
|
|
|
|
Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
|
|
|
|
Snapshot(max_size=config.num_snapshots), trigger=(1, 'epoch'))
|
|
|
|
# print(trainer.extensions)
|
|
|
|
|
|
|
|
trainer.run()
|
|
|
|
trainer.run()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|