datasets

main_custom.py

@@ -203,8 +203,8 @@ def run(epochs, eval_inter, save_inter, model, data_train, data_val, data_test,
             if e % eval_inter == 0:
                 eval(model, data_val, e, criterion_rt, criterion_intensity, metric_rt, metric_intensity, forward,
                      wandb=wandb)
-            if e % save_inter == 0:
-                save(model, 'model_common_' + str(e) + '.pt')
+            # if e % save_inter == 0:
+            #     save(model, 'model_common_' + str(e) + '.pt')
         save_pred(model, data_val, forward, output, file=file)