![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
Chaossll
这个作者很懒,什么都没留下…
展开
-
【Pytorch】学习笔记(训练代码框架)
Pytorch 学习笔记在Pytorch中,Tensor 是包含 data 和 grad 两个部分的,data 放的是真实的数据,而grad中保存的是 计算产生的梯度用于反向传播更新参数使用由 loss.backward()所计算的梯度会产生累计,所以在每次参数更新后需要设置 optim.zero_grad来将上一次的梯度清空pytorch的训练代码主要分为四个模块Prepare Dataimport torchfrom torchvision import transformsfro原创 2021-01-09 10:45:45 · 2428 阅读 · 0 评论 -
【Pytorch】计算图片均值和方差
计算图片数据集的均值和方差因为 pytorch 的 transforms.Normalize 方法需要输入数据集的每个通道的均值和标准差,而网上许多方法都是直接把整个数据集读入内存中进行计算,对于小数据集这样做问题不大,但是对于稍大一点的数据集就可能会出现 Momery Error 即内存不足的问题。故本文采用对数据集进行多次抽样分别计算每次抽样的均值和标准差,最后再求平均的方式来近似计算整个数据集的均值和标准差。import numpy as npimport cv2import osfrom原创 2020-12-28 16:02:19 · 2380 阅读 · 1 评论