Skip to content

Commit

Permalink
Normalize the codes.
Browse files Browse the repository at this point in the history
  • Loading branch information
mtjhl authored Apr 14, 2023
1 parent c4d3b75 commit 6b9f5f4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion yolov6/core/engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,7 @@ def train_before_loop(self):
use_dfl=self.cfg.model.head.use_dfl,
reg_max=self.cfg.model.head.reg_max,
iou_type=self.cfg.model.head.iou_type,
fpn_strides=self.cfg.model.head.strides)
fpn_strides=self.cfg.model.head.strides)

if self.args.fuse_ab:
self.compute_loss_ab = ComputeLoss_ab(num_classes=self.data_dict['nc'],
Expand Down

0 comments on commit 6b9f5f4

Please sign in to comment.