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 dtype log when default (None) is used w/o AMP

Browse Source
This commit is contained in:
Ross Wightman 2025-01-07 11:47:22 -08:00
parent 92f610c982
commit 1969528296
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -597,7 +597,7 @@ def main():
_logger.info('Using native Torch AMP. Training in mixed precision.') _logger.info('Using native Torch AMP. Training in mixed precision.')
else: else:
if utils.is_primary(args): if utils.is_primary(args):
_logger.info(f'AMP not enabled. Training in {model_dtype}.') _logger.info(f'AMP not enabled. Training in {model_dtype or torch.float32}.')
# optionally resume from a checkpoint # optionally resume from a checkpoint
resume_epoch = None resume_epoch = None

2
validate.py
View File

@ -192,7 +192,7 @@ def validate(args):
amp_autocast = partial(torch.autocast, device_type=device.type, dtype=amp_dtype) amp_autocast = partial(torch.autocast, device_type=device.type, dtype=amp_dtype)
_logger.info('Validating in mixed precision with native PyTorch AMP.') _logger.info('Validating in mixed precision with native PyTorch AMP.')
else: else:
_logger.info(f'Validating in {model_dtype}. AMP not enabled.') _logger.info(f'Validating in {model_dtype or torch.float32}. AMP not enabled.')
if args.fuser: if args.fuser:
set_jit_fuser(args.fuser) set_jit_fuser(args.fuser)