Pytorch
1
coasxu
Paint Cooler Code!
展开
-
nn.BCELoss总结
nn.BCELoss总结本章内容nn.BCELossnn.BCEWithLogitsLoss本章内容版本pytorch 1.0nn.BCELoss用于计算预测值和真实值之间的二元交叉熵损失(Binary Cross Entropy)主要用于多标记任务(一个样本对应多个类别)调用时输入参数:Input: (N,∗)(N, *)(N,∗), ∗*∗表示可有任意个数的额外维度。(已经过Sigmoid)Target:(N,∗)(N, *)(N,∗), 与Input的维度相同原创 2020-09-24 19:40:50 · 1851 阅读 · 0 评论 -
nn.CrossEntropyLoss总结
CrossEntropyLossCrossEntropyLoss包含了nn.LogSoftmax() nn.NLLLoss()原创 2020-09-24 19:21:43 · 1313 阅读 · 0 评论 -
Win10下使用tensorboard
Win10下使用tensorboard快速安装tensorflow,选择1.12版本,速度块。pip install tensorflow==1.12.0启动tensorboardtensorboard --logdir=. --port=6701注意这一步返回的链接可能是TensorBoard 1.12.2 at http://DESKTOP-KAKDOWKA:6701 ...原创 2020-04-16 15:17:37 · 266 阅读 · 0 评论 -
Pytorch张量函数记录
Pytorch 常用函数记录1. Tensor相关函数主要包含torch.xxx各类函数,这些函数主要用于对torch.Tensor进行操作。torch.norm(范数)torch.norm(input, p, dim, out=None) → Tensor介绍:用于计算Tensor的p范数参数:input (Tensor) – 输入张量p (int, float, inf, -inf, ‘fro’, ‘nuc’, optional) – 范数计算中的幂指数值或字符串>>原创 2020-07-15 17:25:11 · 267 阅读 · 0 评论 -
Pytorch.clamp:将小于0的元素修改为0,截断元素的取值空间
Pytorch.clamp:将小于0的元素修改为0,截断元素的取值空间torch.clamp(input, min, max, out=None) → Tensor将输入中每个元素夹紧到[min,max]区间中,算法公式如下:yi={min if xi<minxi if min≤xi≤maxmax if xi>...原创 2020-04-18 13:13:48 · 12119 阅读 · 1 评论 -
Pairwise-ranking loss代码实现对比
Pairwise-ranking loss代码在Pairwise-ranking loss中我们希望正标记的得分都比负标记的得分高,所以采用以下的形式作为损失函数。其中c+c_+c+是正标记,c−c_{-}c−是负标记。J=∑i=1n∑j=1c+∑k=1c−max(0,1−fj(xi)+fk(xi))J=\sum_{i=1}^{n} \sum_{j=1}^{c_{+}} \sum_{k...原创 2020-04-18 11:34:26 · 10856 阅读 · 4 评论 -
pytorch1.0版faster rcnn配置
pytorch1.0版faster rcnn配置参考 记pytorch版faster rcnn配置运行中的一些坑环境Python3.6pytorch 1.0GTX 1080CUDA 8.0项目地址 https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0 【注意,我选用的是pytorch-1.0分支】安装方...原创 2020-03-14 15:37:16 · 206 阅读 · 0 评论 -
配置faster-rcnn-pytorch出现的问题
配置faster-rcnn-pytorch出现的问题配置https://github.com/jwyang/faster-rcnn.pytorchbranch1.01. AttributeError: module ‘scipy.misc’ has no attribute 'imread’问题因为scipy中已经移除了这个方法,要么使用imageio.imread来代替,要么就安装sci...原创 2020-03-07 11:07:37 · 403 阅读 · 0 评论 -
Pytorch切换GPU或CPU
Pytorch切换GPU或CPUPytorch版本:>=0.40Pytorch默认是在cpu上训练,我们可以通过to(device)方法切换到GPUto(device)参数to(device)方法接受的参数为torch.device类型。可以写为device = torch.device("cuda:0") # 使用gpu0我习惯这种写法,如果你系统中没有可用的cuda,直接切...原创 2020-02-29 13:06:29 · 8012 阅读 · 0 评论 -
Google Colab 保存和恢复模型(Pytorch)
在Google Colab上,我们可以直接训练深度学习模型,接下来介绍一下Pytorch如何保存和恢复模型。首先,将你的Google Drive挂载上。点击Mount Drive运行这段代码(上一步完成后,会自动生成的)在文件夹中就能看到已挂载的Drive文件夹...原创 2019-12-18 22:31:25 · 13142 阅读 · 13 评论 -
torchtext ValueError: Requested tokenizer basic_english
使用google colab来运行Pytorch官网上的Transformer例子使用"basic_english"作为分割函数,代码如下。import torchtextfrom torchtext.data.utils import get_tokenizerTEXT = torchtext.data.Field(tokenize=get_tokenizer("basic_englis...原创 2019-12-18 21:09:24 · 802 阅读 · 0 评论