oops

pytorch · Feb 16, 2023 · 035ccd7 · 035ccd7
1 parent 9f1738b
commit 035ccd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/references/detection/train.py b/references/detection/train.py
@@ -281,7 +281,7 @@ def main(args):
     for epoch in range(args.start_epoch, args.epochs):
         if args.distributed:
             train_sampler.set_epoch(epoch)
-        # train_one_epoch(model, optimizer, data_loader, device, epoch, args.print_freq, scaler)
+        train_one_epoch(model, optimizer, data_loader, device, epoch, args.print_freq, scaler)
         lr_scheduler.step()
         if args.output_dir:
             checkpoint = {