通过对比上面两种方法,我们可以总结出它们的区别: load方法会直接加载整个模型的结构和参数,适用于模型结构不变的情况。 load_state_dict方法只加载模型的参数字典,需要手动将参数与模型结构对应起来,适用于只加载参数的情况。 因此,当我们只需要加载模型参数而不需要加载整个模型结构时,推荐使用load_state_dict方法。 ...
1. 而load_state_dict是net的一个方法 是将torch.load加载出来的数据加载到net中