如何将自己的数据集用于火炬数据集类？

Question

我有自己的数据集的numpy变量X ad y。如何在pytorch数据类中使用它？

我看过一些教程，但是我仍然不确定如何从数据集中设置数据扩充和迭代器（或批处理提取器）。

为此目的有在线教程吗？

Answer 1

我强烈建议您阅读PyTorch提供的Dataset类。

这是官方PyTorch示例的link。另外，这是example，显示了PyTorch Dataset类的深入操作以及完整的实现。