Update train.py

nikitabelooussovbtis · Jan 9, 2023 · 0a674a9 · 0a674a9
1 parent a67eb89
commit 0a674a9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -282,6 +282,7 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
     hyp['label_smoothing'] = opt.label_smoothing
     model.nc = nc  # attach number of classes to model
     model.hyp = hyp  # attach hyperparameters to model
+    print(dataset.labels)
     model.class_weights = labels_to_class_weights(dataset.labels, nc).to(device) * nc  # attach class weights
     model.names = names