changdazhou преди 1 година
родител
ревизия
806e4760e9
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      paddlex/configs/structure_analysis/PicoDet_layout_1x.yaml

+ 1 - 1
paddlex/configs/structure_analysis/PicoDet_layout_1x.yaml

@@ -15,7 +15,7 @@ CheckDataset:
     val_percent: null
 
 Train:
-  num_classes: 4
+  num_classes: 11
   epochs_iters: 50
   batch_size: 24
   learning_rate: 0.4