add le drop

SlongLiu · SlongLiu · commit 497d182ce8a2 · 2022-04-13T14:40:35.000+08:00
diff --git a/main.py b/main.py
@@ -348,7 +348,7 @@ def check_keep(keyname, ignorekeywordlist):
                     'args': args,
                 }, checkpoint_path)
 
-
+        lr_scheduler.step()
         if args.output_dir:
             checkpoint_paths = [output_dir / 'checkpoint.pth']
             # extra checkpoint before LR drop and every 100 epochs
