Closed
Description
Changing the DeepLab ONNX export script to use the new dynamo-based exporter, by adding dynamo=True
to the torch.export
call, produces an ONNX model which rten-convert
cannot convert. This issue tracks the various features which are missing to make it work.
- Support for
Constant
operators which store their value in an attribute other thanvalue
(eg.value_ints
,value_floats
). See spec (Supportvalue_{int, ints, float, floats}
attributes inConstant
op #649) - Support
CastLike
operator (ImplementCastLike
operator #646) - Support for
start
andend
attributes inShape
operator (Supportstart
andend
attributes forShape
operator #648) - Support for
pytorch_half_pixel
value forcoordinate_transformation_mode
attribute ofResize
operator (Supportpytorch_half_pixel
value forcoordinate_transform_mode
#647) - Support for function operators (eg.
_aten_native_batch_norm_inference_onnx
in this model). Update: This requirement goes away if the model is optimized after export. See Add option to use dynamo ONNX exporter in export-timm-model.py #651. After fixing that a subsequent issue arises where inference produces different results than for TorchScript. - Investigate different inference result when comparing optimized Dynamo-exported model against TorchScript-exported model (Fix
pytorch_half_pixel
resize mode in Resize op #654)
Metadata
Metadata
Assignees
Labels
No labels