Browse Source

[Fix] Change save_interval_epochs of ppyolo to 5

Bobholamovic 3 năm trước cách đây
mục cha
commit
343f646f7d
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  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,