diff --git a/train.py b/train.py index 1de70dcc..4bb68399 100755 --- 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()])))