Remove bn-tf arg

pull/1239/head
Ross Wightman 2 years ago
parent 0012bf7fb5
commit 4c8bb295ab

@ -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,

Loading…
Cancel
Save