mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix updated validation_batch_size fallback

Browse Source
This commit is contained in:
Ross Wightman 2021-09-02 14:44:53 -07:00
parent 5db057dca0
commit 0639d9a591
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -539,7 +539,7 @@ def main():
loader_eval = create_loader( loader_eval = create_loader(
dataset_eval, dataset_eval,
input_size=data_config['input_size'], input_size=data_config['input_size'],
batch_size=args.validation_batch_size, batch_size=args.validation_batch_size or args.batch_size,
is_training=False, is_training=False,
use_prefetcher=args.prefetcher, use_prefetcher=args.prefetcher,
interpolation=data_config['interpolation'], interpolation=data_config['interpolation'],