More epochs in FCCDN tutorial

2 years ago · 3147e26dc1
parent e48418fb25
commit 3147e26dc1
1 changed files with 2 additions and 2 deletions
--- a/tutorials/train/change_detection/fccdn.py
+++ b/tutorials/train/change_detection/fccdn.py
@ -78,11 +78,11 @@ model = pdrs.tasks.cd.FCCDN()
 # 执行模型训练
 model.train(
-    num_epochs=5,
+    num_epochs=10,
    train_dataset=train_dataset,
    train_batch_size=4,
    eval_dataset=eval_dataset,
-    save_interval_epochs=2,
+    save_interval_epochs=4,
    # 每多少次迭代记录一次日志
    log_interval_steps=50,
    save_dir=EXP_DIR,