From e8a03809731062e5dbfa1263c99bb3a1174578f3 Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Mon, 9 Sep 2024 13:54:34 +0200
Subject: [PATCH] dataset
---
main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.py b/main.py
index 936e040..97cecf3 100644
--- a/main.py
+++ b/main.py
@@ -272,7 +272,7 @@ def main_rt(args):
data_sources=[args.dataset_train, args.dataset_train, args.dataset_train])
else:
data_train, data_val, data_test = load_data(batch_size=args.batch_size, n_train=args.n_train, n_test=args.n_test,
- data_sources=[args.dataset_train,args.dataset_test,args.dataset_train])
+ data_sources=[args.dataset_train,args.dataset_train,args.dataset_test])
print('\nData loaded')
# if args.model == 'RT_self_att' :
# model = RT_pred_model_self_attention()
--
GitLab