Merge pull request #2216 from Jackwaterveg/develop_fix

[Bug fix] fix bug for benchmark
pull/2229/head
TianYuan 2 years ago committed by GitHub
commit 09487a0e11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -389,6 +389,7 @@ class DataLoaderFactory():
config['mini_batch_size'] = args.ngpu
config['subsampling_factor'] = 1
config['num_encs'] = 1
config['shortest_first'] = False
elif mode == 'valid':
config['manifest'] = config.dev_manifest
config['train_mode'] = False

Loading…
Cancel
Save