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

Merge pull request #244 from hollance/master

Browse Source 
Bug fix: test_time_pool would be set to a non-False value
This commit is contained in:
Ross Wightman 2020-10-02 15:53:48 -07:00 committed by GitHub
commit 186075ef03
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference.py
View File

@ -73,7 +73,7 @@ def main():
(args.model, sum([m.numel() for m in model.parameters()])))
config = resolve_data_config(vars(args), model=model)
model, test_time_pool = model, False if args.no_test_pool else apply_test_time_pool(model, config)
model, test_time_pool = (model, False) if args.no_test_pool else apply_test_time_pool(model, config)
if args.num_gpu > 1:
model = torch.nn.DataParallel(model, device_ids=list(range(args.num_gpu))).cuda()

2
validate.py
View File

@ -139,7 +139,7 @@ def validate(args):
_logger.info('Model %s created, param count: %d' % (args.model, param_count))
data_config = resolve_data_config(vars(args), model=model)
model, test_time_pool = model, False if args.no_test_pool else apply_test_time_pool(model, data_config)
model, test_time_pool = (model, False) if args.no_test_pool else apply_test_time_pool(model, data_config)
if args.torchscript:
torch.jit.optimized_execution(True)