![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Python
平凡搬砖
这个作者很懒,什么都没留下…
展开
-
pytorch反向传播遇到的坑,持续积累
pytorch 反向传播遇到的坑原创 2022-06-20 15:56:08 · 1223 阅读 · 0 评论 -
一点关于OrderedDict
前言记录一下OrderedDict是因为pytorch加载参数(torch.load(<model_path>)['state_dict'])是OrderedDict类型,其中以(key, value)的形式存储着整个模型的参数。作为Dictionary的一个子类,最大不同在于保存了items的插入顺序。具体参考:文档。直接上例子测试代码:from collections import OrderedDictod = OrderedDict()od['a'] = 123od['b原创 2021-10-13 19:10:17 · 1139 阅读 · 0 评论 -
Python,记录 stack, vstack, hstack 区别
Python | 记录 stack, vstack, hstack 区别1. stack2. vstack3. hstack1. stackstack的两个对象必须形状相同,结果会向外扩展一个维度,例:import numpy as npa = np.array([[1,2,3,4], [2,3,4,5]])b = np.array([[2,3,4,5],[5,6,7,8]])print(a.shape, a)print(b.shape, b)c = np.stack((a,b), axis原创 2021-05-04 21:37:04 · 154 阅读 · 0 评论