mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Remove debug print from train.py
This commit is contained in:
parent
59ec7e6a53
commit
fd9061dbf7
1
train.py
1
train.py
@ -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()])))
|
||||||
|
Loading…
x
Reference in New Issue
Block a user