diff --git a/examples/rs_research/train_cd.py b/examples/rs_research/train_cd.py index f0c0410..f737ef5 100644 --- 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, # 每多少次迭代记录一次日志