From ce236ba65869e8a06f17bbdb4f6878c9f97e93bb Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 21 Oct 2024 15:59:15 +0200
Subject: [PATCH] datasets
---
main_custom.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/main_custom.py b/main_custom.py
index 7048332..a8db447 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)
--
GitLab