@@ -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,
# 每多少次迭代记录一次日志
Bobholamovic