pytorch
解决方案
对conv2d等网络层初始化一下。
# 默认方法
for m in model.modules():
if isinstance(m, (nn.Conv2d, nn.Linear)):
nn.init.xavier_uniform_(m.weight)
初始化方式可以参考:https://blog.csdn.net/weixin_38314865/article/details/106049511
pytorch
解决方案
对conv2d等网络层初始化一下。
# 默认方法
for m in model.modules():
if isinstance(m, (nn.Conv2d, nn.Linear)):
nn.init.xavier_uniform_(m.weight)
初始化方式可以参考:https://blog.csdn.net/weixin_38314865/article/details/106049511