csdn_wmk-CSDN博客

原创获取tensor大小

记录：tensorflow中获取tensor大小yourTensor.get_shape().as_list() 直接输出即可。参考https://www.cnblogs.com/huangshiyu13/p/7531085.html

2021-12-17 17:03:52 1767

原创 pdb命令

1、pdb常用命令break / b 设置断点continue / c 继续执行程序list / l 查看当前代码段step / s 进入函数return / r 执行代码直到从当前函数返回exit / q 中止并退出next / n 执行下一行pp

2021-11-14 19:44:21 641

原创 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 702

原创 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 3002

原创 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 206

原创 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 724 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 138 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 173 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 99

原创 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 204

原创 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 314 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 112 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 177

csdn_wmk的博客