Update train.py

nikitabelooussovbtis · Jan 9, 2023 · b29d142 · b29d142
1 parent 0a674a9
commit b29d142
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -282,7 +282,8 @@ 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)
+    print("\n\n\n\n dataset labels")
+    print(dataset.labels, flush=True)
     model.class_weights = labels_to_class_weights(dataset.labels, nc).to(device) * nc  # attach class weights
     model.names = names