print_checkpoint_saving

main_sphere_compression.py

@@ -336,6 +336,7 @@ def save_checkpoint(state_dics, is_best, output_folder, filename=None):
         output_fileAddr = os.path.join(output_folder, f"checkpoint_{state_dics['epoch']:03d}")
     output_fileAddr += ".pth.tar"
     torch.save(state_dics, output_fileAddr)
+    print(f'saved checkpoint to {output_fileAddr}')
     if is_best:
         fp_best = os.path.join(output_folder, 'checkpoint_best_loss.pth.tar')
         print(f'saved best model to {fp_best}')