为什么要使用collate_fn. 这里先从dataset的运行机制讲起. 在dataloader按照batch进行取数据的时候, 是取出大小等同于batch size的index列表; ...
確定! 回上一頁