pytorch具体如何载入自定义的数据库
时间: 2023-03-07 17:34:25 浏览: 107
pytorch 自定义数据集加载方法
5星 · 资源好评率100%
PyTorch提供了一系列的数据加载工具,可以很容易地载入自定义的数据库。通常,可以使用 torch.utils.data.DataLoader 类来载入数据,并使用 torch.utils.data.TensorDataset 类将数据转换为 PyTorch 的 Tensor 格式。
阅读全文
pytorch 自定义数据集加载方法