ColossalAI
ColossalAI copied to clipboard
Trying to adpat DLRM but encounter api incompatibility
Describe the feature
DLRM is taking multiple input of different shape as default dataloader design. When trying to adapt it using engine api, I meet a
Can we allow tuple input for this case as it need less effort compared to turning it into dict?
Can you provide more information? A code snippet will be more helpful.
Hi, I understand your question. This dict setting is designed for pipeline and we will support tuple in later versions.
We have updated a lot. This issue was closed due to inactivity. Thanks.