pytorch mobile trial with a self-trained gan(U-GAT-IT,pytorch version) , to make a person looks more beautiful
Training and test datasets are generated by styleGANv2
inut_tensor = torch.rand(1,3,256,256)
mobile = torch.jit.trace(processedModel,inut_tensor)
mobile.save("test2.pt")
processedModel is your model file and loaded by pytorch.