Przeglądaj źródła

[Fix] Change save_interval_epochs of ppyolo to 5

Bobholamovic 3 lat temu
rodzic
commit
343f646f7d
1 zmienionych plików z 1 dodań i 1 usunięć
  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,