No, try to run the model without any exports first.
Once it’s working properly with the transform method as well as with the direct model call (p_wct(cont_img, ...)), which calls the forward method, you could start trying to export it.
If your model isn’t working in eager mode first, the export would probably raise unhelpful errors.