问题描述
我已经通过 PytorchLightning
训练了一个 Trainer.fit
模型并保存了生成的模型。现在我想加载模型并运行单轮验证,无需任何培训。最好的方法是什么? API 似乎只提供 Trainer.fit
和 Trainer.test
。我可以将 MyLightningModule.test_step
覆盖为 MyLightningModule.validation_step
并运行 Trainer.test
,但这感觉就像一个黑客。或者,我可以使用 Trainer.fit
运行 limit_train_batches=1,max_epochs=1
,但这也感觉像是一种黑客攻击。最惯用的方法是什么?
理想情况下,我希望获得一个漂亮而干净的日志,而不用不必要的条目污染它(Trainer.fit
解决方法就是这种情况)。
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)