|
|
@ -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'],
|
|
|
|