From 4118896b204b614fb6e1d8a7a897749cd3bd41ef Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 25 Sep 2024 09:38:06 +0200
Subject: [PATCH] transfer setup
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index 283c0fc..ae2ec09 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -185,7 +185,7 @@ def run(epochs, eval_inter, save_inter, model, data_train, data_val, data_test,
wandb=wandb)
if e % save_inter == 0:
save(model, 'model_common_' + str(e) + '.pt')
- save_pred(model, data_val, 'rt', output)
+ save_pred(model, data_val, 'both', output)
else :
for e in range(1, epochs + 1):
--
GitLab