请教一下,下面的代码该怎么修改呀?谢谢哈

482 次浏览0个评论2021年07月24日

小白一枚,学习pytorch中,遇到个问题。 网上查了,说是nn.sigmoid()改为了 torch.sigmoid(input) , 请问一下,下面的代码该怎么改呀?

class LogisitcRegression(nn.Module):

def __init__(self, D_in):
    super(LogisitcRegression, self).__init__()
    self.linear = nn.Linear(D_in, 1)
    self.sigmoid = nn.sigmoid()

def forward(self, x):
    x = self.linear(x)
    output = self.sigmoid(x)
    return output

Lr_model = LogisitcRegression(2)

nn.sigmoid()改为了torch.sigmoid(),结果报错了。说是缺少一个参数。

提交评论

请登录后评论

用户评论

  • 2021-07-27 14:08:23
    已经解决了,谢谢。
    0 赞 0 条评论 回复
    评论
    查看更多评论!
  • 2021-07-27 14:08:23
    已经解决了,谢谢。
    0 赞 0 条评论 回复
    评论
    查看更多评论!

更多相关好文

    当前暂无更多相关好文推荐...

    公告提示

    Pytorch是Facebook的AI研究团队发布了一个Python工具包,是Python优先的深度学习框架。作为numpy的替代品;使用强大的GPU能力,提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。