Pytorch Lightning 踩坑记录
最近两周开始了解和上手学习pytorch lightning,这个框架,csdn和知乎资料都比较少,而且框架相对年轻,还是有不少该改进的地方
lightning 可以和torch 兼容,在某些方面反而有冲突,可能需要自己写callback 函数实现
今天记录一下自己的踩坑(不分先后)
模型test 无输出
正确代码
def test_step(self, batch, batch_idx):
x, y = batch
logits = self.forward(x)
preds =
原创
2021-07-07 19:18:44 ·
1777 阅读 ·
1 评论