paddle的dataloader中的collate_fn参数
收藏
datald = DataLoader(data,batch_size=32,collate_fn=collate_fn,shuffle=True,
drop_last=True,
num_workers=0)
print(datald.__len__)是batch数量是10个,在collate_fn中print的batch数据迭代只有3次,
for i, batch in enumerate(datald()):
print(i)
print(batch.src)
打印不出i,和batch.src,也不报错,反正就是什么都没有
0
收藏
请登录后评论