pytorch-lightning-template
pytorch-lightning-template copied to clipboard
预测结果保存在哪儿呢?
如题
模板会自行生成一个文件夹lightning_logs
,如果这是你跑的第N个version(相同的模型跑了几次就会有几个version文件夹),那么你的最优模型将会是像这样:ROOT_PATH\lightning_logs\version_0\checkpoints\best-epoch=10-mpsnr=57.62-mssim=0.984.ckpt
希望能帮到你
请问如何用跑完的最优模型预测呢?
同问,这个框架下最佳的inference workflow是怎么样的呢
请问如何用跑完的最优模型预测呢? +1
他们其实想问Trainer.test()或者Trainer.validate()怎么用,尤其test.step()和validate_step(),怎么设计?
他们其实想问Trainer.test()或者Trainer.validate()怎么用,尤其test.step()和validate_step(),怎么设计?
我猜他们想问这一套测试和验证的方法,因为你的main(),只涉及到fit()训练