Fix typo

2 years ago · e4ddab56cd
parent b75e0d19d5
commit e4ddab56cd
1 changed files with 1 additions and 1 deletions
--- a/examples/rs_research/train_cd.py
+++ b/examples/rs_research/train_cd.py
@ -92,7 +92,7 @@ model.train(
    num_epochs=50,
    train_dataset=train_dataset,
    train_batch_size=8,
-    eval_dataset=eval_dataset,
+    eval_dataset=val_dataset,
    # 每多少个epoch验证并保存一次模型
    save_interval_epochs=5,
    # 每多少次迭代记录一次日志