mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix `use_amp` 

Fix https://github.com/rwightman/pytorch-image-models/issues/881
pull/882/head
Shoufa Chen 2021-09-26 12:32:22 +08:00 committed by GitHub
parent 3d9c23af87
commit 908563d060
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -397,7 +397,7 @@ def main():
# setup synchronized BatchNorm for distributed training
if args.distributed and args.sync_bn:
assert not args.split_bn
if has_apex and use_amp != 'native':
if has_apex and use_amp == 'apex':
# Apex SyncBN preferred unless native amp is activated
model = convert_syncbn_model(model)
else:
@ -451,7 +451,7 @@ def main():
# setup distributed training
if args.distributed:
if has_apex and use_amp != 'native':
if has_apex and use_amp == 'apex':
# Apex DDP preferred unless native amp is activated
if args.local_rank == 0:
_logger.info("Using NVIDIA APEX DistributedDataParallel.")