I currently have a basic jupyter notebook that follows this article. My hope is to later convert the onnx file into a tflite file in the future.
However, after running the code linked in the article, I receive a CUDA out of memory error by running the code below. I also ran this on a Google Colab and produced the same error, so I can assume that hardware is not the issue here.
torch.onnx.export(model, input_batch, '../model/deeplab_model_pytorch.onnx', verbose=True)

Does anybody know why this is causing an error?