yjhhhhh__的博客

原创 记录一个关于linear网络参数载入的问题

在使用torch.load和xx.load_state_dict时，对于linear网络加载我只加载了linearnet.weight，然后发输出一直有问题，debug了一晚上后我突然想到，这个网络似乎定义了bias，所以需要将linearnet.bias也一起加载上去。非常微小的错误，但是对于结果影响非常大。