pytorch神经网络第一次网络返回全0有警告怎么办？

Question

这里这个警告怎么消除，因为第一次网络返回的全是0，后面就不是0了？

Answer 1

这是由于PyTorch已经将nn.init.xavier_uniform函数废弃，并推荐使用nn.init.xavier_uniform_函数。这个警告并不会导致网络出现错误，但是在训练神经网络时可能会对性能产生一定的影响。

为了解决这个问题，可以将nn.init.xavier_uniform替换为nn.init.xavier_uniform_，具体代码如下：

# 使用nn.init.xavier_uniform初始化权重
# 替换前：
nn.init.xavier_uniform(model.weight)
# 替换后：
nn.init.xavier_uniform_(model.weight)

在使用nn.init.xavier_uniform_函数时，注意添加下划线，否则仍然会出现警告。通过这种方式可以避免警告，并提高神经网络的性能。