The WeGO-Torch project is currently in an early access state and might encounter a few known usage issues. Here are the details of these issues and the necessary steps to address them:
- WeGO-Torch cannot support RCNN models (with control-flow)
because:
- RCNN Model Support: WeGO-Torch does not currently support RCNN models with control-flow due to dynamic shape problems. In RCNN models, the shape of tensors can change during runtime when different images are provided as inputs. This poses challenges for deployment in WeGO. To make RCNN models compatible with WeGO, manual modifications are required to eliminate this constraint.
- Input Type Compatibility: RCNN models often accept
Tensor[]
as an input type, which WeGO's compile API does not support. Additionally, usingTensor []
as the input type implies that the float model itself is batch-sensitive, and the quantized models obtained through tracing differ based on the batch size used during the TorchScript tracing phase. To deploy these models in WeGO, the following steps are recommended:- Replace
Tensor []
withTensor
orTensor, Tensor, ...
(when the number of inputs is known) as the input type in the original float model. - The batch size used for inference in WeGO must be the same as the one used in the export phase during quantization.
- Replace
- WeGO-Torch currently covers only a subset of operators that data center DPUs can support. Consequently, certain operators might be dispatched to the CPU for execution, even if data center DPUs can support them.