FoBaokuku 2022-04-23 16:47 采纳率: 100%
浏览 67
已结题

data.DataLoader参数batch_size是作用于dataset里的哪一个函数?

问题遇到的现象和发生背景

自建数据集MiddleDataset,在进行实例化后:train_set = MiddleDataset(config.dataset_path, transform_train, 'train')
注意:数据集MiddleDataset里边有基本的__init__;getimet;和__len__三个方法
不太理解下面这句话的意思:self.train_loader = data.DataLoader(dataset=train_set, batch_size=config.batch_size, shuffle=True)
主要不理解的一点是:data.DataLoader参数batch_size是作用于dataset里的哪一个函数?自己在设定batch_size=16后,输出的一个batch有很多个,步态理解这一块?

我想要达到的结果

希望可以透彻理解这一块参数的作用对象。

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 5月1日
    • 创建了问题 4月23日