batch_size的探索

女爷i 2024-04-01 12:43 64阅读 0赞

## **问题** ##

test\_loader 中的y 表示每一个batch对应的128张图片对应的数字，torch.Size(\[256\])表示什么意思？

![18957d44844f499fb5ec91a5a5f6293d.png][]

![389be20f8dd396989af922ec74955483.png][]

##  ##

## **方法** ##

<table> 
 <tbody> 
  <tr> 
   <td>在打印了X的长度之后，发现X的长度也为256，这表示此处用作测试的X是由256个1x28x28的矩阵构成的多元组矩阵集合。也即，y的长度为256，而不是128。<br><img alt="71e84635e3c23b32bfb7f30166cecc81.png" src="https://img-blog.csdnimg.cn/img_convert/71e84635e3c23b32bfb7f30166cecc81.png"><br> for X, y in test_loader: &nbsp;<br> &nbsp; &nbsp; print(X.shape, y.shape)<br> &nbsp; &nbsp; print(y)<br> &nbsp; &nbsp; print(len(X))<br> print(X)<br> 原因：<br> 在初次设置test_loader的batch_size为256，而不是128.<br><img alt="aa97253f5450b86d4eb3fe1e5ab431c5.png" src="https://img-blog.csdnimg.cn/img_convert/aa97253f5450b86d4eb3fe1e5ab431c5.png"></td> 
  </tr> 
 </tbody> 
</table>

##  ##

## **结语** ##

在本次探索中，通过print()，我对test\_loader中的（X, y）的数据格式有了一定的认识，同时对batch\_size的重要性有了一定的了解。

[18957d44844f499fb5ec91a5a5f6293d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/9a8b1b11086c420cb379a7bebb5e43e8.png
[389be20f8dd396989af922ec74955483.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/01/9d1da2889a6c472185fa846474153013.png

发表评论取消回复

表情：

评论列表（有 0 条评论，64人围观）

还没有评论，来说两句吧...

相关阅读

相关 CNN基础——如何设置BatchSize

目录 1、什么是BatchSize 2、为什么需要 Batch\_Size？ 3、如何设置Batch\_Size 的值？ 4、在合理范围内，增大Batch\_Siz

港控/mmm°/ 2023年01月20日 13:52/ 0 赞/ 4 阅读

相关 openvino在不同batchsize下的性能对比

之前写过一篇关于tensorrt的：[tensorrt在不同batchsize下的性能对比][tensorrt_batchsize] 模型还是使用的这篇文章的模型，只是将其转

今天药忘吃喽~/ 2023年01月10日 14:58/ 0 赞/ 146 阅读

相关训练过程--Batchsize（总之一般情况下batchsize能设多大就设多大）

只做记录，原文：https://blog.csdn.net/wydbyxr/article/details/84855489 尽可能地设大batchsize 在合理范围

ゝ一世哀愁。/ 2023年01月09日 12:42/ 0 赞/ 131 阅读

相关 tensorrt在不同batchsize下的性能对比

tensorrt利用GPU进行加速，天然的GPU是适合并行计算，因此加大batchsize是优化tensorrt常见的方式之一 tensorrt默认是batchsize=1，

骑猪看日落/ 2023年01月08日 07:27/ 0 赞/ 249 阅读

相关深度学习中学习率和batchsize对模型准确率的影响

> 本内容来自其他的人解析，参考链接在最后的注释。 1. 前言目前深度学习模型多采用批量随机梯度下降算法进行优化，随机梯度下降算法的原理如下: ![在这里插入图片

雨点打透心脏的1/2处/ 2022年12月19日 13:16/ 0 赞/ 297 阅读

相关 iteration, batchsize, epoch定义、区别和联系

iteration：数据进行一次前向-后向的训练，就是更新一次参数； batchsize：每次迭代（iteration）训练图片的数量； epoch：1个epoch就

布满荆棘的人生/ 2022年05月24日 11:40/ 0 赞/ 190 阅读

相关 yield的一些探索

yield的一些探索文章目录 yield的一些探索开头概念 yield 与 return 生成器对

╰+攻爆jí腚メ/ 2022年04月03日 18:44/ 0 赞/ 220 阅读

相关神经网络2. epoch， iteration， batchsize相关理解和说明

本文为原创文章转载请注明出处：http://blog.csdn.NET/qq\_20259459 和作者信息。原文超链接（点击阅读原文） batchsize：中文翻译为批

以你之姓@/ 2022年03月25日 11:48/ 0 赞/ 184 阅读

相关神经网络中Epoch、Iteration、Batchsize辨析

1.深度学习中经常看到epoch、iteration和batchsize，三者区别：（1）batchsize：批大小。在深度学习中，一般采用SGD训练，即每次训练在训练集

谁借莪１个温暖的怀抱￠/ 2022年02月28日 08:22/ 0 赞/ 226 阅读

相关深度学习中的epoch、 iteration、batchsize和dropout的含义

1.batchsize（批大小）：在深度学习中，一般采用stochastic gradient descent（SGD）随机梯度下降训练，即每次训练在训练集中取batc

Love The Way You Lie/ 2022年02月18日 13:45/ 0 赞/ 396 阅读