1. 首页
  2. 编程语言
  3. Python
  4. pytorch Dataset,DataLoader产生自定义的训练数据案例

pytorch Dataset,DataLoader产生自定义的训练数据案例

上传者: 2022-03-10 05:33:05上传 PDF文件 143.78 KB 热度 14次

但这会出现一个问题,当样本总数train_data_nums与batch_size不能整取时,最后一个batch会少于规定batch_size的大小,比如这里样本总数train_data_nums=10,batch_size=7,第一次迭代会产生7个样本,第二次迭代会因为样本不足,只能产生3个样本。

下载地址
用户评论