PyTorch-网络的创建,预训练模型的加载
PyTorch-网络的创建,预训练模型的加载
本文是PyTorch使用过程中的的一些总结,有以下内容:
构建网络模型的方法
网络层的遍历
各层参数的遍历
模型的保存与加载
从预训练模型为网络参数赋值
主要涉及到以下函数的使用
add_module,ModulesList,Sequential 模型创建
modules(),named_modules(),children(),na...
转载
2020-04-01 11:49:21 ·
183 阅读 ·
0 评论