![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
技术解决方法
weixin_45073393
这个作者很懒,什么都没留下…
展开
-
Pytorch之softmax和log_softmax的区别
函数 Softmax(x) 也是一个 non-linearity, 但它的特殊之处在于它通常是网络中一次操作. 这是因为它接受了一个实数向量并返回一个概率分布.其定义如下. 定义 x 是一个实数的向量(正数或负数都无所谓, 没有限制). 然后, 第i个 Softmax(x) 的组成是 输出是一个概率分布: 每个元素都是非负的, 并且所有元素的总和都是1. ...原创 2021-09-13 10:57:17 · 133 阅读 · 0 评论 -
pytorch之dataloader深入剖析
DataLoader本质上就是一个iterable(跟python的内置类型list等一样),并利用多进程来加速batch data的处理,使用yield来使用有限的内存 pytorch 的数据加载到模型的操作顺序是这样的: ① 创建一个 Dataset 对象 ② 创建一个 DataLoader 对象 ③ 循环这个 DataLoader 对象,将img, label加载到模型中进行训练 DataLoader是PyTorch中数据读取的一个重要接口,该接口定义在dataloader.py中,只..原创 2021-09-13 10:32:58 · 130 阅读 · 0 评论