0
点赞
收藏
分享

微信扫一扫

pytorch train模式

脱下愤怒的小裤衩 2022-04-25 阅读 54
pytorch

前传判断:

    def forward(self, x): # [3,112,112]

        if self.training:
            print("train")
        else:
            print("eval")

调用:


if __name__ == "__main__":
    net = MFN_85m()

    net.eval()
举报

相关推荐

0 条评论