From 3147e26dc1f76d9381c4c16388b18449854479e3 Mon Sep 17 00:00:00 2001 From: Bobholamovic Date: Mon, 5 Sep 2022 14:15:50 +0800 Subject: [PATCH] More epochs in FCCDN tutorial --- tutorials/train/change_detection/fccdn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials/train/change_detection/fccdn.py b/tutorials/train/change_detection/fccdn.py index 62abbba..318fa0e 100644 --- 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,