Pytorch
Vincent__Lai
这个作者很懒,什么都没留下…
展开
-
【Pytorch】快速输出网络模型信息方法
首先,安装torchsummarypip isntall torchsummary下面是demo代码,其中(3, 100, 100)分别为单张图片的通道数、高、宽from torchsummary import summaryimport torchvisionimport torchdef model_info(model): device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') back原创 2020-12-28 10:51:36 · 1751 阅读 · 1 评论 -
【Pytorch】 Attempting to deserialize object on a CUDA device but torch.cuda.is_available()
torch.loads原创 2020-08-18 10:50:21 · 2674 阅读 · 0 评论 -
【Pytorch】RuntimeError: one of the variables needed for gradient computation has been modified by
参考:RuntimeError: one of the variables needed for gradient computation has been modified by an inplace o报错RuntimeError: one of the variables needed for gradient computation has been modified by an in...原创 2020-04-06 15:35:36 · 742 阅读 · 0 评论 -
【Pytorch】torch.gather函数作用
例如,二维的情况:假如gather中选择的dim为0,则索引为行。即:输出结果的第i行第j列是原始数据中第j列第?行的值其中?由index中第i行第`j列元素决定i,j从0开始import torcht = torch.Tensor([[1,2,3],[4,5,6]])index = torch.LongTensor([[0,1,1],[1,0,0]])print(t)pr...原创 2019-11-24 19:44:53 · 924 阅读 · 0 评论 -
【Python】如何自定义对象被for遍历时返回的内容
一个对象如果想被for循环作为一个可迭代的对象遍历的话,那么在定义类的时候必要有属性__getitem__(self, index)先看下面的例子:class enmuerate_test(object): def __init__(self): self.a = [1,2,3,4,5] def __getitem__(self, index): ...原创 2019-11-23 22:35:30 · 859 阅读 · 0 评论 -
【Pytroch】Ubuntu下通过graphviz实现网络可视化方法
安装graphvizpip install graphvizsudo apt install graphvizdot -version若安装成功则会显示:Using device: dot:dot:coreThe plugin configuration file:/usr/lib/x86_64-linux-gnu/graphviz/config6awas successful...原创 2019-11-14 16:10:28 · 543 阅读 · 0 评论 -
【Pytorch】常用基本结构
load dataimport torch.utils.data as Datatorch_dataset = Data.TensorDataset(x, y) #x,y为tensor形式loader = Data.DataLoader( dataset=torch_dataset, # torch TensorDataset format batch_size=B...原创 2019-11-07 11:03:18 · 607 阅读 · 0 评论 -
【目标检测】Win10+TITAN RTX+CUDA10.0+pytorch1.1搭建CenterNet环境
CenterNet有两个,本文的CenterNet对应的论文是:Objects as Points对应Github地址:xingyizhou/CenterNetUbuntu16.04安装步骤点击这里点击这里下面这两篇博客也对我帮助很大:在SeaShips数据集上训练CenterNet网络(绝对详细)CenterNet训练自己的数据(pytorch0.4.1)环境安装https://g...原创 2019-10-22 15:03:19 · 1859 阅读 · 3 评论 -
【目标检测】Ubuntu16.04+RTX2070+CUDA10.0+pytorch1.1搭建CenterNet环境
环境安装https://github.com/xingyizhou/CenterNet/issues/404原创 2019-10-22 12:04:14 · 4719 阅读 · 14 评论