store checkpoint of initialization
@@ -143,6 +143,8 @@ class TrainingJob(TrainingOrEvaluationJob):
raise Exception(
f"{self.__class__.__name__} was initialized for forward only. You can only call run_epoch()"
)
+ if self.epoch == 0:
+ self.save(self.config.checkpoint_file("init"))
self.config.log("Starting training...")
checkpoint_every = self.config.get("train.checkpoint.every")
