allow_resume_more_epochs

main_sphere_compression.py

@@ -509,9 +509,13 @@ def main():
         optimizer_aux.load_state_dict((checkpoint["optimizer_aux_state_dict"]))
 
         list_mean_losses = checkpoint["list_mean_losses"]
+        epoch_delta = args.max_epochs - len(list_mean_losses)
+         # allow for resuming training with more epochs
+        if epoch_delta > 0: list_mean_losses.extend([None] * (epoch_delta))
 
         if "list_mean_losses_validation" in checkpoint:
             list_mean_losses_validation = checkpoint["list_mean_losses_validation"]
+            if epoch_delta > 0: list_mean_losses.extend([None] * (epoch_delta))
 
         if not args.no_scheduler:
             if "scheduler_state_dict" in checkpoint: