Explorar o código

modify RT-DETR's engine

yjh0410 hai 1 ano
pai
achega
089e7f4f6f
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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