DataLoader for PyTorch, or a tf.data.Dataset for Tensorflow. ... return self.collate_fn([self.get() for _ in range(batch_size)]).
確定! 回上一頁