Hey @raghuramank100 I have saved the model correctly but I want to use it in pytorch so we must know the definition of model then we can load the state_dict from the saved model file.
But what I have is definition of model without the fusion of layer and that’s where the definition of model changing and I can’t load model.
So Do I need to change the model definition according to the fused layers?