From c35bd9f90e580d2933c8855cc807dff9888295c0 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Thu, 26 Sep 2024 17:56:44 +0200
Subject: [PATCH] fix err dataloader main
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index cd1538d..716029c 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -216,7 +216,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,args.dataset_val,args.dataset_test],
+ data_train, data_val, data_test = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_val,args.dataset_test],
batch_size=args.batch_size, length=25)
elif args.forward == 'transfer':
--
GitLab