Преглед на файлове

modify RT-DETR's engine

yjh0410 преди 1 година
родител
ревизия
089e7f4f6f
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      utils/solver/lr_scheduler.py

+ 1 - 1
utils/solver/lr_scheduler.py

@@ -51,7 +51,7 @@ def build_lr_scheduler(cfg, optimizer, resume=None):
     elif cfg['lr_scheduler'] == 'cosine':
         pass
         
-    if resume is not None:
+    if resume is not None and resume.lower() != "none":
         print('keep training: ', resume)
         checkpoint = torch.load(resume)
         # checkpoint state dict