Converting SwinUNETR to ONNX or Tensorflow

[vasudharao08]

Hi, I am trying to convert the model to ONNX using torch.onnx.export() function but script gets 'Killed' abruptly which I am guessing is the memory error. I also looked at nobuco to convert it to Keras but that gives me a weird error:

TypeError: _InstanceNorm.__init__() missing 1 required positional argument: 'num_features'

Python version: 3.10
Torch version: 2.1.1

Any tips? Was someone successfully able to convert the model to ONNX or Tensorflow?
Thank you!