小编jBl*_*ess的帖子

加载 .npy 文件作为 pytorch 的数据集

我在 .npy 文件中对数据进行了预处理,我们将原始数据称为 X.npy,将标签称为 Y.npy。它们的组织方式是匹配两个文件中的每个元素(X 中的第一个元素具有 Y 中的第一个标签等)。如何使用将其加载为数据集torch.utils.data.DataLoader?我对 pytorch 很陌生,任何帮助都会很有用。

python serialization numpy pytorch

8
推荐指数
1
解决办法
6086
查看次数

标签 统计

numpy ×1

python ×1

pytorch ×1

serialization ×1