本文介绍了在PyTorch中遇到的两个常见问题：1) 对于ReLU激活层，需要实例化才能使用，如`relu = nn.ReLU()`，然后应用到输入上。2) Softmax函数的隐式维度选择已弃用，应指定维度，如`softmax = nn.Softmax(dim=0)`。这两个修改能避免警告并确保代码的正确运行。
1）Unable to get repr for <class 'torch.nn.modules.activation.ReLU'>
（大概意思是不能得到对Relu层的描述）
修改之前：
nn.Relu(input)对于激活层，这有个打代码细节，需要对其实例化。
修改之后：
relu = nn.Relu()
output = relu(input)2）UserWarning: Implicit dimension choice for softmax has been deprecated.
大概意思是softmax的隐式维度选择已被弃用。
修改之前：
softmax = nn. Softmax()
output = softmax(input)Softmax()的参数在目前版本python中不手动设置会爆出警告，最好自己设置。
（详细参数作用看
扫一扫
8453
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
