之前看到好几个Pytorch版本的代码,虽然也实现了读取数据组成task, ... batch = self.collate_fn([self.dataset[i] for i in indices]) # Dataset
確定! 回上一頁