csdn_wmk-优快云博客

原创获取tensor大小

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

2021-12-17 17:03:52 1847

原创 pdb命令

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

2021-11-14 19:44:21 810

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

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

原创 Pytorch学习（十）---- 卷积神经网络 CNN

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 代码实现： import os # third-party library import torch import torch.nn as nn import torch.utils.data as Data import torchvision import matplotlib.pyplot .

2021-01-07 23:37:12 266

原创 Pytorch学习（九）---- 优化器

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 优化器学习代码： import torch import torch.utils.data as Data import torch.nn.functional as F import matplotlib.pyplot as plt import os os.environ["KMP_DUPLI.

2020-12-28 11:06:19 833 2

原创 Pytorch学习（八）---- 批数据训练

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 import torch import torch.utils.data as Data BATCH_SIZE = 5 if __name__ == '__main__': x = torch.linspace(1, 10, 10) y = torch.linspace(10, 1.

2020-12-25 19:38:28 197 1

原创 Pytorch学习（七）---- 保存提取

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 import torch from torch.autograd import Variable import matplotlib.pyplot as plt import os os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE" # import os 这里是.

2020-12-19 14:39:09 251 1

原创 Pytorch学习（六）---- 神经网络快速搭建

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 import torch import torch.nn.functional as F from torch.autograd import Variable import matplotlib.pyplot as plt import os os.environ["KMP_DUPLICATE_.

2020-12-18 10:37:25 128

原创 Pytorch学习（五）---- Classification 分类

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 import torch import torch.nn.functional as F from torch.autograd import Variable import matplotlib.pyplot as plt import os os.environ["KMP_DUPLICATE_L.

2020-12-17 20:08:42 252

原创 Pytorch学习（四）---- Regression 回归

莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 学习搭建一个简单的神经网络 import torch from torch.autograd import Variable import torch.nn.functional as F import matplotlib.pyplot as plt import os os.environ["K.

2020-12-16 21:58:42 390 1

原创 Pytorch学习（三）----Activation function

激励函数—Activation function 莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578

2020-12-16 16:19:48 170

原创 Pytorch学习（二）

Variable变量学习莫烦python视频学习笔记视频链接https://www.bilibili.com/video/BV1Vx411j7kT?from=search&seid=3065687802317837578 variable与tensor的不同 import torch from torch.autograd import Variable tensor = torch.FloatTensor([[1, 2], [3, 4]]) variable = Variable(te

2020-12-10 14:04:43 168 1

原创 Pytorch学习（一）

1、Numpy与Pytorch的数据转换 import torch import numpy as np np_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 135 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 281