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

Remove debug print from train.py

Browse Source
This commit is contained in:
Ross Wightman 2021-01-04 12:16:06 -08:00
parent 59ec7e6a53
commit fd9061dbf7
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -332,7 +332,6 @@ def main():
scriptable=args.torchscript, scriptable=args.torchscript,
checkpoint_path=args.initial_checkpoint) checkpoint_path=args.initial_checkpoint)
print(model)
if args.local_rank == 0: if args.local_rank == 0:
_logger.info('Model %s created, param count: %d' % _logger.info('Model %s created, param count: %d' %
(args.model, sum([m.numel() for m in model.parameters()]))) (args.model, sum([m.numel() for m in model.parameters()])))