torch
andyzou221312
这个作者很懒,什么都没留下…
展开
-
dqn利用gym的例子
利用gym构建的dqn例子原创 2022-08-29 19:41:45 · 356 阅读 · 1 评论 -
CBOW中关于 NLLLoss 的含义
学习CBOW的时候对NLLLoss的一点研究:NLLLoss的定义是一大堆公式, 没有仔细学习, 用实践法研究一下其作用:criterion = nn.NLLLoss()…loss = criterion(out, target)在本例中, 其中 out 是一个数组(list), target 是一个float张量out 的长度代表了 字典的长度.print(out)tensor([[-6.7951, -3.5001, -6.4829, -5.4005, -6.5381, -3.1883,原创 2022-05-07 21:39:38 · 833 阅读 · 0 评论 -
torch显示保存拼接数据集图片
先显示一个大的拼接的图片再保存部分小图片import torchfrom torch import nn, optimimport torch.nn.functional as Ffrom torch.autograd import Variablefrom torch.utils.data import DataLoaderfrom torchvision import transformsfrom torchvision import datasetsimport matplotlib原创 2022-04-28 21:42:53 · 1334 阅读 · 1 评论 -
pytorch-beginner-master 填坑学习笔记
04-Convolutional Neural Network 出现tensorflow的问题, 注释掉LOGGER的代码.loss.data[0] 报错, 这是版本问题, 将所有的data[0]替换成item()后成功.image = Variable(image, volatile=True) 报错, 替换成with torch.no_grad(): image = Variabl...原创 2020-02-03 02:53:59 · 293 阅读 · 3 评论