miracleyoo / pytorch-lightning-template

An easy/swift-to-adapt PyTorch-Lighting template. 套壳模板,简单易用,稍改原来Pytorch代码,即可适配Lightning。You can translate your previous Pytorch code much easier using this template, and keep your freedom to edit all the functions as well. Big-project-friendly as well. No need to rewrite your config in hydra.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

预测结果保存在哪儿呢?

wentixiaogege opened this issue · comments

如题

模板会自行生成一个文件夹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()训练