![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Python备忘
天青月白
这个作者很懒,什么都没留下…
展开
-
Pytorch保存和加载训练好的模型,以及加载中提示Can‘t get attribute ‘XXX‘ on <module ‘__main__‘ from ‘YYY‘>的解决方法
保存,分仅保存参数,和保存整个模型1 仅保存参数,轻便,但需把网络模型先执行一遍方法百度2 保存整个模型,无脑全存保存torch.save(rnn, R'D:\Python\11111.pkl')加载.rnn = torch.load(R'D:\Python\11111.pkl')如果直接执行,报错AttributeError: Can't get attribute 'RNN' on <module '__main__' from '<input>原创 2022-03-11 11:31:15 · 3985 阅读 · 0 评论 -
pytorch中将tensor转换为numpy时报错
测试集中,对x预测,想把预测结果装进y这个ndarray中y = rnn(x).to(device)出错TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.参考网上,原因大概是无法直接将cuda tensor转换为ndarray,得经过cpu解决方法:改为y = rnn(x).to(device原创 2022-03-10 22:22:03 · 1136 阅读 · 0 评论 -
Python(Numpy)打印最后若干个数字,以及合并两个ndarray
例如a = np.array([1,2,3,4,5])b = np.ones(5)若打印a的最后3个数字3 4 5,则可以下方式二选一c = print(a[len(a)-3 :len(a)])c = print(a[-3 :len(a)])若将a和b合并成两行,第一行a第二行b(类似matlab的[a;b]),则使用d = np.row_stack((a,b))若是合并成两列,即左是a的列向量,右是b的列向量,有下面两种方法。注意,生成的e中,是划分成[1,0] [2,0原创 2022-03-04 11:41:52 · 1142 阅读 · 0 评论