继承PyTorch DataLoader类

Question

我们可以继承DataLoader类吗？若然，是否有具体的限制？

我知道我们可以为Dataset类这样做，但我需要具体了解DataLoader。

对于特定的用例，我有一些需求，包括覆盖一些DataLoader方法和添加一些新的方法。

真的很感激你的帮助！谢谢..。

Answer 1

谢谢@Jan和@TheodorPeifer为您提供了有用的输入. :)

我能够通过将dataset __get_item__调用中的__get_item__返回与"worker_specific_info"(torch.utils.data.get_worker_info())以及自定义collate_fn实现相结合来解决这个问题。