2019年8月15日
pytorch 中Dataloader中的collate_fn参数
摘要： 一般的，默认的collate_fn函数是要求一个batch中的图片都具有相同size（因为要做stack操作），当一个batch中的图片大小都不同时，可以使用自定义的collate_fn函数，则一个batch中的图片不再被stack操作，可以全部存储在一个list中，当然还有对应的label，如下面 阅读全文
posted @ 2019-08-15 21:06 outthinker 阅读(11100) 评论(0) 推荐(1)