sangyun884 / HR-VITON

Official PyTorch implementation for the paper High-Resolution Virtual Try-On with Misalignment and Occlusion-Handled Conditions (ECCV 2022).
831 stars 171 forks source link

About training settings #7

Closed HITRainer closed 2 years ago

HITRainer commented 2 years ago

Hi, thanks for sharing your codes. But I was confused by the difference between your codes and the description in your paper during retraining your model. For example, when training try-on generator, the code will train the model for 300K iters but it is only 100k mentioned in your paper (Sec 4.1, last line in Page8). So, are the excellent results in your paper trained according to the settings mentioned in the paper or according to the released code settings. And which one should us follow to get your paper results.

sangyun884 commented 2 years ago

Follow the description in the paper and feel free to early stop the training whenever needed.