|
|
@@ -813,7 +813,7 @@ class RTCTrainer(object):
|
|
|
self.optimizer, self.start_epoch = build_yolo_optimizer(self.optimizer_dict, model, args.resume)
|
|
|
|
|
|
# ---------------------------- Build LR Scheduler ----------------------------
|
|
|
- self.lr_scheduler, self.lf = build_lr_scheduler(self.lr_schedule_dict, self.optimizer, args.max_epoch - args.no_aug_epoch)
|
|
|
+ self.lr_scheduler, self.lf = build_lr_scheduler(self.lr_schedule_dict, self.optimizer, args.max_epoch)
|
|
|
self.lr_scheduler.last_epoch = self.start_epoch - 1 # do not move
|
|
|
if self.args.resume and self.args.resume != 'None':
|
|
|
self.lr_scheduler.step()
|
yjh0410