pytorch
开开心心 everyday
这个作者很懒,什么都没留下…
展开
-
PyTorch 学习率调整的几个方法
https://blog.csdn.net/u011995719/article/details/89486359原创 2019-12-19 16:41:53 · 366 阅读 · 0 评论 -
Pytorch 加载、查看预训练模型参数、使用部分预训练模型参数初始化网络(以层为单位按需初始化)
一:查看预训练模型的参数(以Alexnet为例):path = 'I:/迅雷下载/alexnet-owt-4df8aa71.pth'pretrained_dict = torch.load(path)for k, v in pretrained_dict.items(): # k 参数名 v 对应参数值 print(k)运行结果: ...原创 2019-12-18 17:14:45 · 10335 阅读 · 3 评论