ds-nat-j

I've seen other stylegan2 models used in the original demo like stylegan2_human_v2_512.pkl, stylegan2-old-car-config-f.pt etc. If we could parse the model path for these newer models that might come in the...

`python3 test_generator.py --occlusion --test_name 'test1' --tocg_checkpoint "/dbfs/nathan/VTON-test/mtviton.pth" --gen_checkpoint "/dbfs/nathan/VTON-test/gen.pth" --datasetting unpaired --dataroot "/dbfs/nathan/VTON-test" --data_list "/dbfs/nathan/VTON-test/test_pairs.txt"` Error message: ``` File "/databricks/driver/HR-VITON/test_generator.py", line 278, in main() File "/databricks/driver/HR-VITON/test_generator.py", line 272, in main...