From 2030649b0f3b8efece6273457d123ebe9e92073e Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 24 Sep 2024 13:49:32 +0200
Subject: [PATCH] transfer setup
---
main_custom.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index 06b4fbd..7e6d34a 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -202,9 +202,14 @@ 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, data_val, data_test = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_val,args.dataset_test],
+ 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,
+ path_val=args.dataset_val,
+ path_test=args.dataset_test,
+ batch_size=args.batch_size, length=25, pad = True, convert=True, vocab='iapuc')
+
print('\nData loaded')
model = Model_Common_Transformer(encoder_ff=args.encoder_ff, decoder_rt_ff=args.decoder_rt_ff,
--
GitLab