随笔分类 -  torch

摘要：#三个核心函数 torch.save() torch.load() torch.nn.Module.load_state_dict() 状态字典定义 状态字典本质上就是普通的python字典。 对于具有可学习参数的网络层来说，状态字典的键就是网络层，值就是对应的参数张量。 大概如下图所示，网络层的可 阅读全文