torch.save(module.state_dict(), path) :使用 module.state_dict() 函数获取各层已经训练好的参数和缓冲区,然后将参数和缓冲区保存到 path 所指定的 ...
確定! 回上一頁