2025年5月23日
torch-lightning:Trainer.predict()返回值与predict_step()的关系
摘要： 你的问题非常好，涉及 PyTorch Lightning 的 predict_step 和 trainer.predict() 的行为。 1. predict_step 的作用 predict_step 是你在 LightningModule 里定义的每个 batch 的预测逻辑。 你可以返回任意你 阅读全文
posted @ 2025-05-23 17:39 Gold_stein 阅读(114) 评论(0) 推荐(0)