I tried to run test_virtual_tryon.py with GMM model checkpoint provided, but it turned out that there are a lot of unexpected keys in source state_dict and missing keys in source state_dict. So I checked out both GMM and TOM models and figured out that they have same state dicts with same weights in it. I'm not sure this is how it's supposed to be.
I tried to run test_virtual_tryon.py with GMM model checkpoint provided, but it turned out that there are a lot of unexpected keys in source state_dict and missing keys in source state_dict. So I checked out both GMM and TOM models and figured out that they have same state dicts with same weights in it. I'm not sure this is how it's supposed to be.![image](https://user-images.githubusercontent.com/88214807/128363074-d2c27266-5bcc-4249-afc4-4a2a0cd6ebc0.png)