Explorar o código

use cosine lr decay for RT-DETR-R50

yjh0410 hai 1 ano
pai
achega
133cfb673c
Modificáronse 1 ficheiros con 0 adicións e 1 borrados
  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()