baidu_20965275的博客

当我们花费大量的精力训练完网络，下次预测数据时不想再（有时也不必再）训练一次时，这时候torch.save(),torch.load()就要登场了。 保存和加载模型参数有两种方式： 方式一：   torch.save(net.state_dict(),path): 功能：保存训练完的网络的各层参数（即weights和bias) 其中：net.state_dict()获取各层参数，pat...