利用Pytorch DataLoader中的collate_fn自定义批加载方式
摘要：
Pytorch DataLoader中的collate_fn允许用户灵活的定义批样本的加载方式，例如加载变尺寸的张量。 collate_fn (callable, optional) – merges a list of samples to form a mini-batch of Tensor( 阅读全文
posted @ 2020-06-15 22:31 那抹阳光1994 阅读(1347) 评论(0) 推荐(0)
浙公网安备 33010602011771号