Can celeba_distill be converted to torch mobile ?

bryandlee / animegan2-pytorch

PyTorch implementation of AnimeGANv2

MIT License

4.41k stars 641 forks source link

Can celeba_distill be converted to torch mobile ? #81

Open davidjdseo opened 2 years ago

davidjdseo commented 2 years ago

Great work! I'd like to convert your pretrained model, celeba_distill, to pytorch mobile. from torch.utils.mobile_optimizer import optimize_for_mobile Do you know the model specifics that will allow a conversion to the lite interpreter for mobile?