Explorar el Código

revert tutorials

jiangjiajun hace 5 años
padre
commit
7659a3bb41
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      tutorials/train/object_detection/ppyolo.py

+ 1 - 1
tutorials/train/object_detection/ppyolo.py

@@ -50,7 +50,7 @@ model = pdx.det.PPYOLO(num_classes=num_classes)
 model.train(
     num_epochs=270,
     train_dataset=train_dataset,
-    train_batch_size=1,
+    train_batch_size=8,
     eval_dataset=eval_dataset,
     learning_rate=0.000125,
     lr_decay_epochs=[210, 240],