diff --git a/train.py b/train.py index 0440f551..6e74f67b 100755 --- a/train.py +++ b/train.py @@ -213,8 +213,6 @@ parser.add_argument('--drop-block', type=float, default=None, metavar='PCT', help='Drop block rate (default: None)') # Batch norm parameters (only works with gen_efficientnet based models currently) -parser.add_argument('--bn-tf', action='store_true', default=False, - help='Use Tensorflow BatchNorm defaults for models that support it (default: False)') parser.add_argument('--bn-momentum', type=float, default=None, help='BatchNorm momentum override (if not None)') parser.add_argument('--bn-eps', type=float, default=None, @@ -424,7 +422,6 @@ def setup_train_task(args, dev_env: DeviceEnv, mixup_active: bool): drop_path_rate=args.drop_path, drop_block_rate=args.drop_block, global_pool=args.gp, - bn_tf=args.bn_tf, bn_momentum=args.bn_momentum, bn_eps=args.bn_eps, scriptable=args.torchscript,