From 9bd9edf82985b9ad4e2b76e27fcd425315884ff3 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 23 Oct 2024 12:54:16 +0200
Subject: [PATCH] datasets
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index a8db447..c0cdf04 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -222,7 +222,7 @@ def main(args):
data_train, data_val = common_dataset.load_data(path_train=args.dataset_train,
path_val=args.dataset_val,
path_test=args.dataset_test,
- batch_size=args.batch_size, length=args.seq_length, pad = False, convert=False, vocab='unmod')
+ batch_size=args.batch_size, length=args.seq_length, pad = True, convert=True, vocab='unmod')
elif args.forward == 'rt':
data_train, data_val = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_val,args.dataset_test],
batch_size=args.batch_size, length=args.seq_length)
--
GitLab