diff --git a/main_custom.py b/main_custom.py index 7048332c805a7e88a84b13704abe9d8b52e78b17..a8db44702d0a9f7d04949170d50a68cc70496d01 100644 --- a/main_custom.py +++ b/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)