diff --git a/main_custom.py b/main_custom.py index 59f34e834557b4d8b41844ee600e9133b2d5320c..24d11ffa3395934843a16d16297de247c50bef10 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,