fix error if val_start > max_epoch

main_sphere_compression.py

@@ -589,7 +589,7 @@ def main():
     plotGradFolder = os.path.join(args.out_dir, args.foldername_plot_gradient) if args.foldername_plot_gradient is not None else None
     if plotGradFolder:
         os.makedirs(plotGradFolder, exist_ok=True)
-
+    perform_validation = False
     for epoch in range(last_epoch + 1, args.max_epochs): # epoch=0...max_epochs-1, printing and checkpoint saving in 1...max_epochs
         loss_train = train_epoch(train_dataloader, struct_loader, net, criterion, optimizer, optimizer_aux, args.patch_res_train, args.n_patch_per_sample, args.print_freq, epoch, args.clip_max_norm, plotGradFolder)
 
@@ -643,8 +643,8 @@ def main():
             save_checkpoint(states, is_best, args.out_dir, only_best_model=(not save_regular_checkpoint))
 
     # Saving last results
-    is_best = loss_validation["loss"] < best_loss if args.validation_data else False
-    best_loss = min(loss_validation["loss"], best_loss) if args.validation_data else best_loss
+    is_best = loss_validation["loss"] < best_loss if perform_validation else False
+    best_loss = min(loss_validation["loss"], best_loss) if perform_validation else best_loss
     states = {
         'epoch': epoch,
         'net_state_dict': net.state_dict(),