- 博客(15)
- 收藏
- 关注
原创 获取tensor大小
记录:tensorflow中获取tensor大小yourTensor.get_shape().as_list() 直接输出即可。参考https://www.cnblogs.com/huangshiyu13/p/7531085.html
2021-12-17 17:03:52 1753
原创 pdb命令
1、pdb常用命令break / b 设置断点continue / c 继续执行程序list / l 查看当前代码段step / s 进入函数return / r 执行代码直到从当前函数返回exit / q 中止并退出next / n 执行下一行pp
2021-11-14 19:44:21 628
原创 Can‘t pickle local object ‘data.numerize.<locals>.<lambda>‘
困扰了好久,记录一下。报错原因是由于pickle dump 不能dump 带lambda的对象例如代码:def numerize(self, data, user2id, item2id): user_id = map(lambda x: user2id[x], data['user_id']) item_id = map(lambda x: item2id[x], data['item_id']) pickle.dump(user_reviews, open(os
2021-10-28 20:56:11 698
原创 torch_scatter安装
记一下torch_scatter的安装:直接pip总会报错,查询发现需要手动下载包,下载链接:https://pytorch-geometric.com/whl/torch-1.5.0.html在这里面找到和python对应的版本,python3.6下的是torch_scatter-2.0.4+cu102-cp36-cp36m-win_amd64.whl然后再pip,但是一定要注意,要把下载的文件放在自己的安装路径之下,否则会报错Requirement ‘torch_scatter-2.0.4+
2021-04-30 12:07:50 2983
原创 Pytorch学习(十)---- 卷积神经网络 CNN
莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578代码实现:import os# third-party libraryimport torchimport torch.nn as nnimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot .
2021-01-07 23:37:12 205
原创 Pytorch学习(九)---- 优化器
莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578优化器学习代码:import torchimport torch.utils.data as Dataimport torch.nn.functional as Fimport matplotlib.pyplot as pltimport osos.environ["KMP_DUPLI.
2020-12-28 11:06:19 712 2
原创 Pytorch学习(八)---- 批数据训练
莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578import torchimport torch.utils.data as DataBATCH_SIZE = 5if __name__ == '__main__': x = torch.linspace(1, 10, 10) y = torch.linspace(10, 1.
2020-12-25 19:38:28 136 1
原创 Pytorch学习(七)---- 保存提取
莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578import torchfrom torch.autograd import Variableimport matplotlib.pyplot as pltimport osos.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"# import os 这里是.
2020-12-19 14:39:09 164 1
原创 Pytorch学习(六)---- 神经网络快速搭建
莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578import torchimport torch.nn.functional as Ffrom torch.autograd import Variableimport matplotlib.pyplot as pltimport osos.environ["KMP_DUPLICATE_.
2020-12-18 10:37:25 98
原创 Pytorch学习(五)---- Classification 分类
莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578import torchimport torch.nn.functional as Ffrom torch.autograd import Variableimport matplotlib.pyplot as pltimport osos.environ["KMP_DUPLICATE_L.
2020-12-17 20:08:42 199
原创 Pytorch学习(四)---- Regression 回归
莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578学习搭建一个简单的神经网络import torchfrom torch.autograd import Variableimport torch.nn.functional as Fimport matplotlib.pyplot as pltimport osos.environ["K.
2020-12-16 21:58:42 308 1
原创 Pytorch学习(三)----Activation function
激励函数—Activation function莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578
2020-12-16 16:19:48 108
原创 Pytorch学习(二)
Variable变量学习莫烦python视频学习笔记 视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578variable与tensor的不同import torchfrom torch.autograd import Variabletensor = torch.FloatTensor([[1, 2], [3, 4]])variable = Variable(te
2020-12-10 14:04:43 111 1
原创 Pytorch学习(一)
1、Numpy与Pytorch的数据转换import torchimport numpy as npnp_data = np.arange(6).reshape((2,3))torch_data = torch.from_numpy(np_data) # numpy数据转换为torch数据tensor2array = torch_data.numpy() # 再转换成numpy数据print('\nnumpy:', np_data)print('\ntorch:
2020-12-10 14:03:27 94 1
原创 Pytorch的安装问题解决
安装总是出现Unable to create process …的问题解决:直接卸载python编译器,重新下载一个安装命令:pip install torch=1.4.0 torchvision=0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
2020-12-05 11:13:19 175
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人