Remove debug print from train.py

2021-01-04 12:16:06 -08:00 · 2021-01-04 12:16:06 -08:00 · fd9061dbf7
parent 59ec7e6a53
commit fd9061dbf7
1 changed files with 0 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -332,7 +332,6 @@ def main():
        scriptable=args.torchscript,
        checkpoint_path=args.initial_checkpoint)

-    print(model)
    if args.local_rank == 0:
        _logger.info('Model %s created, param count: %d' %
                     (args.model, sum([m.numel() for m in model.parameters()])))