![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
阿二狗
悄悄的打工,然后惊艳所有打工人
展开
-
不懂pytorch里transforms.Compose()函数的同学看过来!
首先compose本身的英文含义有组成、排版、组合的意思。然后看到这里,我们大概可以猜出该函数的作用。(这其实也可以加深我们对该函数的记忆pytorch torchvision transform可以对PIL.Image进行各种变换1.torchvision.transforms.CenterCrop(size)将给定的PIL.Image进行中心切割,得到给定的size,size可以是tuple,(target_height, target_width)。size也可以是一个Integer,在这种情况原创 2021-04-16 15:40:33 · 4719 阅读 · 2 评论 -
nn.reLU(inplace)的理解
from torch import nnrelu=nn.ReLU(inplace=True)inplace参数,默认为False,计算得到的值不会覆盖之前的值,如果设置为True,则会把计算得到的值直接覆盖到输入中,这样可以节省内存/显存。之所以可以覆盖是因为在计算ReLU的反向传播时,秩序根据输出就能够推算出反向传播的梯度,但是只有少数的autograd操作支持inplace,除非你明确知道自己在做什么,否则一般不要使用inplace操作。...原创 2020-12-27 16:41:03 · 3272 阅读 · 0 评论