pytorch
烟火笑风尘
众生皆苦
展开
-
pytorch Parameter类源码
https://github.com/pytorch/pytorch/blob/27027a48041be0931938c4ea057419545003f3f6/torch/nn/parameter.py#L5import torchfrom collections import OrderedDictclass Parameter(torch.Tensor): r"""A ...原创 2019-10-13 16:00:11 · 428 阅读 · 0 评论 -
pytorch module 源码
https://github.com/pytorch/pytorch/blob/27027a48041be0931938c4ea057419545003f3f6/torch/nn/modules/module.py#L137from collections import OrderedDict, namedtupleimport functoolsimport itertools...原创 2019-10-13 14:06:55 · 2758 阅读 · 0 评论 -
pytorch Conv 卷积层源码
https://github.com/pytorch/pytorch/blob/master/torch/nn/modules/conv.py# coding=utf-8import mathimport torchfrom torch.nn.parameter import Parameterfrom .. import functional as Ffrom .. imp...原创 2019-10-13 14:05:51 · 6800 阅读 · 4 评论 -
pytorch插值函数interpolate——图像上采样-下采样,scipy插值函数zoom
在训练过程中,需要对图像数据进行插值,如果此时数据是numpy数据,那么可以使用scipy中的zoom函数:from scipy.ndimage.interpolation import zoomdef zoom(input, zoom, output=None, order=3, mode='constant', cval=0.0, prefilter=True):...原创 2019-10-06 18:51:43 · 11021 阅读 · 1 评论 -
pytorch 执行torch.from_numpy(np.expand(tmp_array,axis=0))时报错
ValueError: some of the strides of a given numpy array are negative. This is currently not supported, but will be added in future releases.此时解决办法是:将tmp_array=tmp_array.copy(),执行通过。...原创 2019-07-21 21:21:12 · 1810 阅读 · 0 评论 -
pytorch DataLoader中的参数:collate_fn函数
def default_collate(batch): r"""Puts each data field into a tensor with outer dimension batch size""" error_msg = "batch must contain tensors, numbers, dicts or lists; found {}" elem_typ...原创 2019-06-30 19:50:14 · 2778 阅读 · 0 评论 -
pytorch DataLoader源码解析
#文章源码来自:pytorch源码。一、DataLoader的源码:class DataLoader(object): r""" Data loader. Combines a dataset and a sampler, and provides single- or multi-process iterators over the dataset. ...原创 2019-06-30 19:45:19 · 3515 阅读 · 0 评论 -
pytorchCNN网络权重
ssh://user2@219.216.65.100:6002/home/user2/anaconda3/envs/chenyafang/bin/python -u /home/user2/yd/lungcancer_0521/DSB2017-master/add_by_yd/test_net_structure.pypreBlock.0.weightpreBlock.0.biaspreB...原创 2019-05-29 20:33:07 · 1254 阅读 · 0 评论 -
pytorch expand()函数
转自https://pytorch-cn.readthedocs.io/zh/latest/package_references/Tensor/#expandsizes1. 返回tensor的一个新视图,单个维度扩大为更大的尺寸。 tensor也可以扩大为更高维,新增加的维度将附在前面。 扩大tensor不需要分配新内存,只是仅仅新建一个tensor的视图,其中通过将stride设为0,一维将...原创 2019-05-17 16:41:24 · 84209 阅读 · 6 评论 -
pytorch 中tensor在CPU和GPU之间转换,以及numpy之间的转换
1. CPU tensor转GPU tensor:cpu_imgs.cuda()2. GPU tensor 转CPU tensor:gpu_imgs.cpu()3. numpy转为CPU tensor:torch.from_numpy( imgs )4.CPU tensor转为numpy数据:cpu_imgs.numpy()5. note:GPU tenso...原创 2019-05-24 16:59:37 · 67533 阅读 · 6 评论 -
判断pytorch是否支持GPU加速
print torch.cuda.is_available()原创 2019-04-30 09:58:18 · 19576 阅读 · 4 评论 -
pytorch中Tensor的数据类型
1. 32位浮点型:torch.FloatTensora=torch.Tensor( [[2,3],[4,8],[7,9]], )print "a:",aprint "a.size():",a.size()print "a.dtype:",a.dtypeb=torch.FloatTensor( [[2,3],[4,8],[7,9]] )print "b:",bprint "b....原创 2019-04-30 09:55:53 · 39310 阅读 · 2 评论 -
pytorch backward机制理解
m=Variable( torch.FloatTensor( [ [2,3] ] ),requires_grad=True )# j=torch.zeros(2,2)# k=Variable( torch.zeros( 1,2 ) )## k[0,0]=m[0,0]**2+3*m[0,1]# k[0,1]=m[0,1]**2+2*m[0,0]## k.backward( torch...原创 2019-01-17 20:46:37 · 1190 阅读 · 0 评论