最近两周开始了解和上手学习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中有一个标量不是曲线,只有一个点太难看,结果就没有输出了
在这里插入图片描述

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐