From 0b45201c3ff324fde8153c59d056c45d544fa1e5 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 24 Sep 2024 14:07:40 +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 78b8bb0..7e6d34a 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -202,7 +202,7 @@ def main(args):
path_test=args.dataset_test,
batch_size=args.batch_size, length=25, pad = True, convert=True, vocab='iapuc')
elif args.forward == 'rt':
- data_train, _, _ = dataloader.load_data(data_sources=[args.dataset_train,'database/data_holdout.csv','database/data_holdout.csv'],
+ data_train, _, _ = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_train,args.dataset_train],
batch_size=args.batch_size, length=25)
_, data_val, data_test = common_dataset.load_data(path_train=args.dataset_val,
--
GitLab