Explorar o código

revert tutorials

jiangjiajun %!s(int64=5) %!d(string=hai) anos
pai
achega
7659a3bb41
Modificáronse 1 ficheiros con 1 adicións e 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],