Pytorch Lightning 踩坑记录
最近两周开始了解和上手学习pytorch lightning,这个框架,csdn和知乎资料都比较少,而且框架相对年轻,还是有不少该改进的地方lightning 可以和torch 兼容,在某些方面反而有冲突,可能需要自己写callback 函数实现今天记录一下自己的踩坑(不分先后)模型test 无输出正确代码def test_step(self, batch, batch_idx):x, y = b
·
最近两周开始了解和上手学习pytorch lightning,这个框架,csdn和知乎资料都比较少,而且框架相对年轻,还是有不少该改进的地方
lightning 可以和torch 兼容,在某些方面反而有冲突,可能需要自己写callback 函数实现
今天记录一下自己的踩坑(不分先后)
模型test 无输出
正确代码
def test_step(self, batch, batch_idx):
x, y = batch
logits = self.forward(x)
preds = torch.argmax(logits, dim=1)
acc = accuracy(preds, y)
f1_score = f1(preds, y, num_classes=self.num_classes)
# self.log("f1", f1_score)
self.log("test_acc", acc,logger=False,on_epoch=True,)
必须logger=True
这样的话也会记录到tensotboard,之前设置为false 就是觉得tensotboard中有一个标量不是曲线,只有一个点太难看,结果就没有输出了
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
已为社区贡献2条内容
所有评论(0)