Explorar o código

[Fix] Change save_interval_epochs of ppyolo to 5

Bobholamovic %!s(int64=3) %!d(string=hai) anos
pai
achega
343f646f7d
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

@@ -84,7 +84,7 @@ model.train(
     train_batch_size=4,
     eval_dataset=eval_dataset,
     # 每多少个epoch存储一次检查点
-    save_interval_epochs=10,
+    save_interval_epochs=5,
     # 每多少次迭代记录一次日志
     log_interval_steps=4,
     save_dir=EXP_DIR,