torch保存权重和参数量大小不一致
问题说明及相关代码
采用param.numel()计算参数
for param in net.parameters():
num_params += param.numel()
结果如下:
采用torch.save()保存权重大小
torch.save(net.cpu().state_dict(), save_path)
结果如下:
网上的部分解释
发现两者的大小关系确实接近四倍,但其中具体的细节不清楚,希望帮忙解答下疑惑,详细解释下,万分感谢。