瀏覽代碼

use cosine lr decay for RT-DETR-R50

yjh0410 1 年之前
父節點
當前提交
133cfb673c
共有 1 個文件被更改,包括 0 次插入1 次删除
  1. 0 1
      engine.py

+ 0 - 1
engine.py

@@ -1297,7 +1297,6 @@ class RTDetrTrainer(object):
                 xi = [0, nw]  # x interp
                 for x in self.optimizer.param_groups:
                     x['lr'] = np.interp(ni, xi, [0.0, x['initial_lr'] * self.lf(self.epoch)])
-                exit()
                                 
             # To device
             images = images.to(self.device, non_blocking=True).float()