Struggling with jit.trace/save on Pytorch Segmentation Models model, any help?

I am currently trying to create a lightweight binary segmentation model for mobile use (pytorch\_mobile in a Flutter App).

Originally I trained a regular UNet model which worked fine, however had a huge file size which made it unrealistic for mobile deployment. Since then I have been trying to find a solution in creating a more low-weight binary segmentation model that is a reasonable file size for mobile development.

I created a Unet model with a mobilenet encoder using the …

computervision pytorch segmentation