Solved by user_123454321 in post #2 
              
              
                Can you try 
import torch.utils.mobile_optimizer as mobile_optimizer
mobile_optimizer.optimize_for_mobile(traced_model)