From fd9061dbf74df86431128ce0ca0547858c7600ca Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 4 Jan 2021 12:16:06 -0800 Subject: [PATCH] Remove debug print from train.py --- train.py | 1 - 1 file changed, 1 deletion(-) 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()])))