Przeglądaj źródła

More epochs in FCCDN tutorial

Bobholamovic 2 lat temu
rodzic
commit
3147e26dc1
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      tutorials/train/change_detection/fccdn.py

+ 2 - 2
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,