![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
ephramliu
NWPU硕士
展开
-
ImportError: Can‘t find framework /System/Library/Frameworks/OpenGL.framework.ImportError:Error occu
pip install pyglet==1.5.11原创 2021-04-02 15:45:44 · 863 阅读 · 0 评论 -
AttributeError: Can‘t get attribute ‘Net‘ on module ‘__main__‘
在使用pytorch深度学习框架的时候,我们加载预先训练好的完整pkl模型时,如果报错:AttributeError: Can't get attribute 'Net' on module '__main__'此时我们应该在主函数内加载一个当时训练时的类,用于初始化神经网络,如下:# 加入:初始化class Net(nn.Module): def __init__(self, hidden_layers=64): super(Net, self).__init__()原创 2021-01-18 15:12:10 · 6302 阅读 · 3 评论 -
【Python】随机种子 random/numpy/pytorch/pytorch.cuda
设立随机种子能够确保随机数的初始值在每次实验时相同;即能确保每次实验处于相同的初始状态;import random, torchimport numpy as npseed = 1random.seed(seed)np.random.seed(seed)torch.manual_seed(seed)torch.cuda.manual_seed(seed)其中 seed = 1 表示编号为 1 的随机值,是一组随机值;也可以换成别的整数数字,比如 123,456 之类的;通常不原创 2020-11-23 10:54:32 · 392 阅读 · 0 评论 -
list numpy array tensor转换
1.1 list 转 numpy.arrayndarray = np.array(list)1.2 numpy 转 listlist = ndarray.tolist()2.1 list 转 torch.Tensortensor=torch.Tensor(list)2.2 torch.Tensor 转 list先转numpy,后转listlist = tensor.numpy().tolist()3.1 torch.Tensor 转 numpyndarray = tensor.numpy原创 2020-10-20 11:38:32 · 723 阅读 · 0 评论 -
Pytorch squeeze() 和 unsqueeze() 方法区别
1 增加维度 unsqueeze()在这里插入代码片2 减少维度 squeeze()在这里插入代码片原创 2020-10-18 23:29:31 · 78 阅读 · 0 评论 -
【Pytorch】保存神经网络模型
1 只保存模型参数# 保存torch.save(model.state_dict(), './parameter.pkl')# 加载model = TheModelClass(...)model.load_state_dict(torch.load('./parameter.pkl'))2 保存完整模型# 保存torch.save(model, './model.pkl')# 加载model = torch.load('./model.pkl')...原创 2020-10-17 20:28:09 · 1167 阅读 · 0 评论 -
pytorch 向量转化为one-hot编码
#pytorch 向量转化为one-hot编码import torch#原始向量index = torch.tensor([[1], [2], [0], [3]])onehot = torch.zeros(4, 4)onehot.scatter_(1, index, 1)print(onehot)#结果tensor([[0., 1., 0., 0.], [0., 0., 1., 0.], [1., 0., 0., 0.], [0., 0., 0., 1.]])...原创 2020-10-09 16:44:07 · 1597 阅读 · 0 评论 -
pytorch 深入理解 tensor.scatter_ ()用法
pytoch 深入理解 tensor.scatter_ ()用法在 pytorch 库下理解 torch.tensor.scatter()的用法。作者在网上搜索了很多方法,最后还是觉得自己写一篇更为详细的比较好,转载请注明。首先,scatter() 和 scatter_() 的作用是一样的,但是 scatter() 不会直接修改原来的 Tensor,而 scatter_() 会修改原先的 Tensor。1 API格式torch.Tensor.scatter_(dim, index, src) → T原创 2020-08-02 20:22:49 · 11852 阅读 · 6 评论