From acaa97c53312635624b65be7d1006b6aed6fc3ba Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 30 Sep 2024 09:29:30 +0200
Subject: [PATCH] fix save preds with file
---
main_custom.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/main_custom.py b/main_custom.py
index ecbb936..ff5a2b1 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -200,6 +200,7 @@ def run(epochs, eval_inter, save_inter, model, data_train, data_val, data_test,
def main(args):
+ print('file', args.file)
if args.wandb is not None:
os.environ["WANDB_API_KEY"] = 'b4a27ac6b6145e1a5d0ee7f9e2e8c20bd101dccd'
os.environ["WANDB_MODE"] = "offline"
--
GitLab