[Training] The gradient builder has not been registered: ConvTranspose for node

[elephantpanda]

Describe the issue

When trying to create the training artifacts for a model containing a ConvTranspose2d node I get this error.

Using: https://onnxruntime.ai/docs/api/python/on_device_training/training_artifacts.html

Basically I just want it to create a training ONNX containing a ConvTransposeGrad node the same way it creates a ConvGrad node for a Conv2D node.

But I get the error: "The gradient builder has not been registered: ConvTranspose for node"

Anyway to work round this?

P.S. Is there a list of supported/unsupported operators for training artifacts?

(Possibly this may be implemented in 1.16 but when I do pip install onnxruntime.training it installs version 1.15.1)

To reproduce

Try to create training artifacts with a model containing a ConvTranspose2D node.

Urgency

No response

ONNX Runtime Installation

Released Package

ONNX Runtime Version or Commit ID

1.15.1

PyTorch Version

1.13.1+cpu

Execution Provider

Default CPU

Execution Provider Library Version

CPU

[baijumeswani]

ConvTransposeGrad is available for CUDA EP in ORT 1.16 (PR for reference #17201).

The ConvTransposeGrad kernel has not been implemented for CPU EP yet.

[elephantpanda]

ConvTransposeGrad is available for CUDA EP in ORT 1.16 (PR for reference #17201).

The ConvTransposeGrad kernel has not been implemented for CPU EP yet.

Thanks how do I get it? When I do "pip install onnxruntime.training" it gives me version 1.15.1

[baijumeswani]

python -m pip install cerberus flatbuffers h5py numpy>=1.16.6 onnx packaging protobuf sympy setuptools>=41.4.0
pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training

^ on linux with cuda 11.8 support.

Refer to https://onnxruntime.ai/index.html#getStartedTable to get other installation commands.

[github-actions]

This issue has been automatically marked as stale due to inactivity and will be closed in 7 days if no further activity occurs. If further support is needed, please provide an update and/or more details.

[github-actions]

This issue has been automatically closed due to inactivity. Please reactivate if further support is needed.