From 6a145fe0a28cada6312fa91d217b7898f784acc2 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Oct 2024 14:26:34 +0200
Subject: [PATCH] local integration working
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index 59f34e8..24d11ff 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -238,7 +238,7 @@ def main(args):
batch_size=args.batch_size, length=25, pad = True, convert=True, vocab='unmod')
elif args.forward == 'reverse':
- _, data_val, data_test = dataloader.load_data(data_sources=['database/data_holdout.csv',args.dataset_val,args.dataset_test],
+ _, data_val, data_test = dataloader.load_data(data_sources=['database/data_train.csv',args.dataset_val,args.dataset_test],
batch_size=args.batch_size, length=25)
data_train, _, _ = common_dataset.load_data(path_train=args.dataset_train,
--
GitLab