Skip to content
Snippets Groups Projects
Commit c35bd9f9 authored by Schneider Leo's avatar Schneider Leo
Browse files

fix err dataloader main

parent 45eea11f
No related merge requests found
...@@ -216,7 +216,7 @@ def main(args): ...@@ -216,7 +216,7 @@ def main(args):
path_test=args.dataset_test, path_test=args.dataset_test,
batch_size=args.batch_size, length=25, pad = True, convert=True, vocab='iapuc') batch_size=args.batch_size, length=25, pad = True, convert=True, vocab='iapuc')
elif args.forward == 'rt': 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) batch_size=args.batch_size, length=25)
elif args.forward == 'transfer': elif args.forward == 'transfer':
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment