Skip to content

Commit bfc9225

Browse files
authored
fix lint issues
1 parent 4a9545a commit bfc9225

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

references/detection/train.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def get_args_parser(add_help=True):
143143

144144
# Prototype models only
145145
parser.add_argument("--weights", default=None, type=str, help="the weights enum name to load")
146-
146+
147147
# Mixed precision training parameters
148148
parser.add_argument("--amp", action="store_true", help="Use torch.cuda.amp for mixed precision training")
149149

@@ -211,9 +211,9 @@ def main(args):
211211

212212
params = [p for p in model.parameters() if p.requires_grad]
213213
optimizer = torch.optim.SGD(params, lr=args.lr, momentum=args.momentum, weight_decay=args.weight_decay)
214-
214+
215215
scaler = torch.cuda.amp.GradScaler() if args.amp else None
216-
216+
217217
args.lr_scheduler = args.lr_scheduler.lower()
218218
if args.lr_scheduler == "multisteplr":
219219
lr_scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=args.lr_steps, gamma=args.lr_gamma)

0 commit comments

Comments
 (0)