소스 검색

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