Inheriting the PyTorch DataLoader class



我们可以继承DataLoader类吗?如果有,有什么具体的限制吗?
我知道我们可以为Dataset类这样做,但是我需要特别了解DataLoader。

我对一个特定的用例有一些需求,它涉及覆盖一些DataLoader方法并添加一些新方法。

非常感谢您的帮助!谢谢…

感谢@Jan和@TheodorPeifer的有用输入…!:)

我能够通过数据集__get_item__调用的NamedTuple返回与"worker_specific_info"(torch.utils.data.get_worker_info())以及自定义collate_fn实现的组合来解决这个问题。

最新更新