Skip to content

Commit 48b5490

Browse files
authored
Update train.py
1 parent 63fb65a commit 48b5490

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

panoptic/train.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -222,9 +222,9 @@ def train(hyp, opt, device, callbacks): # hyp is path/to/hyp.yaml or hyp dictio
222222

223223
# Model attributes
224224
nl = de_parallel(model).model[-1].nl # number of detection layers (to scale hyps)
225-
hyp['box'] *= 3 / nl # scale to layers
226-
hyp['cls'] *= nc / 80 * 3 / nl # scale to classes and layers
227-
hyp['obj'] *= (imgsz / 640) ** 2 * 3 / nl # scale to image size and layers
225+
#hyp['box'] *= 3 / nl # scale to layers
226+
#hyp['cls'] *= nc / 80 * 3 / nl # scale to classes and layers
227+
#hyp['obj'] *= (imgsz / 640) ** 2 * 3 / nl # scale to image size and layers
228228
hyp['label_smoothing'] = opt.label_smoothing
229229
model.nc = nc # attach number of classes to model
230230
model.hyp = hyp # attach hyperparameters to model

0 commit comments

Comments
 (0)