Feature Request: Add custom data module support to TorchEstimator
Feature Request: Add custom data module support to TorchEstimator: a task in LegoFlow-SWE (Harbor dataset). Problem TorchEstimator currently hardcodes its data loading to use only Petastorm-based readers. This prevents users from supplying alternative data loading strategies, such as…
The task
**Problem** `TorchEstimator` currently hardcodes its data loading to use only Petastorm-based readers. This prevents users from supplying alternative data loading strategies, such as GPU-accelerated loading via NVTabular, which is already supported in the `KerasEstimator` through a `data_module` parameter. We need…
Part of Lego-X/LegoFlow-SWE.