翻译自 Some important PyTorch tasks – A concise summary from a vision researcher 以前喜欢写 Keras，自从接触了 PyTorch，…

实际上faster-rcnn对于输入的图片是有resize操作的，在resize的图片基础上提取feature map，而后generate一定数量的RoI。 我想首先去掉这个resize的操作，对每张图都是在原始图片基…

1 import torch 2 import torch.nn as nn 3 from torch.autograd import Variable 4 import torch.nn.functional as F…

原文： http://www.voidcn.com/article/p-rtzqgqkz-bpg.html 最近看了下 PyTorch 的损失函数文档，整理了下自己的理解，重新格式化了公式如下，以便以后查阅。 注意下面的…

版本：pytorch 1.0 指定GPU显卡 os.environ["CUDA_VISIBLE_DEVICES"] = "1" 设定全局的tensor 初始化为 tensor.cuda() if torch.cuda.i…

前言 最近使用PyTorch感觉妙不可言，有种当初使用Keras的快感，而且速度还不慢。各种设计直接简洁，方便研究，比tensorflow的臃肿好多了。今天让我们来谈谈PyTorch的预训练，主要是自己写代码的经验以及论…

这几天关于accuracy和loss的计算有一些疑惑，原来是自己还没有弄清楚。 给出实例 def train(train_loader, model, criteon, optimizer, epoch): train_…

最近从tensorflow转向pytorch，感受到了动态调试的方便，也感受到了一些地方的不同。 所有实验都是基于uint16类型的单通道灰度图片。 一开始尝试用opencv中的cv.imread读取图片，发现会默认读8…

转自：https://blog.csdn.net/LoseInVain/article/details/86476010 前言： 本文主要介绍在pytorch中的Batch Normalization的使用以及在其中容易…

TH 和 THNN （C语言库）是Pytorch的最底层； 读懂pytorch的python层的代码，能帮助我们理解组装积木搭建模型的过程； 但要了解每块积木的特性和内在构造，就需要读懂pytorch的C/C++层的代码…

前言 最近跑的模型都比较大，尤其是Bert， 这真的是难为我 1080ti 了， 在Bert的Example中，官方提供了一些 Trick 来帮助我们加速训练，很良心， 但感觉还不够，于是花费一些时间整理出一个 Tric…

这是一篇面向新手的博文：因为本人也是新手，记录一下自己在做这个项目遇到的大大小小的坑。 按照下面的例子写就好了 import torch as t from torch.utils import data import …