请教一下,下面的代码该怎么修改呀?谢谢哈
小白一枚,学习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已经解决了,谢谢。
-
微信公众号文章/菜单添加小程序时路径如何获取? 2021-12-22
-
如何轻松获取微信小程序路径path? 2021-12-22
-
cannot import name 'CUDA_HOME' from 'mmcv.utils' 2021-12-05
-
vgg的loss一轮达到ln(1/n)阈值,如何解决 2021-11-21
-
如何下载使用utils库 2021-10-27
热门文章
-
cannot import name 'CUDA_HOME' from 'mmcv.utils' 2021-12-05
-
vgg的loss一轮达到ln(1/n)阈值,如何解决 2021-11-21
-
如何下载使用utils库 2021-10-27
-
pytorch调用geforce rtx 3060 2021-10-24
-
pytorch中如何实现多模型的并联 2021-09-15
栏目最新文章
Pytorch是Facebook的AI研究团队发布了一个Python工具包,是Python优先的深度学习框架。作为numpy的替代品;使用强大的GPU能力,提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。公告提示
- pytorch中文文档
- pytorch官方文档
其他产品
-
提交评论