I am trying to quantize the CenterNet Resnet-50 using AIMET, but the QuantSim() API breaks after 6 hours without any errors/warnings.
Model Link: https://github.com/xuannianz/keras-CenterNet
Also I am not able to create ONNX for this model using tf2onnx package, got the warnings saying some custom ops are not supported
The lamba functions are defined in the model (keras-CenterNet/resnet.py at 39cb123a94d7774490df28e637240de03577f912 · xuannianz/keras-CenterNet · GitHub).
Can you give me some pointers on how to implement these custom ops to pass through AIMET? Thanks in advance!