我們可以繼承 DataLoader 類嗎?如果是這樣,是否有任何具體限制?
我知道我們可以為 Dataset 類這樣做,但我需要特別了解 DataLoader。
我對特定用例有一些要求,其中涉及覆寫一些 DataLoader 方法并添加一些新方法。
非常感謝您的幫助!謝謝....
uj5u.com熱心網友回復:
感謝@Jan 和@TheodorPeifer 提供有用的意見..!:)
NamedTuple
我能夠通過將資料集__get_item__
呼叫的回傳與“worker_specific_info”( torch.utils.data.get_worker_info()
) 以及自定義collate_fn
實作相結合來解決此問題。
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/496817.html
上一篇:具有多型移動賦值運算子和記憶體安全的std::move
下一篇:僅與內部實作的介面