transfer setup

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,