search

fnzhan / UNITE

[CVPR 2022 Oral] Marginal Correspondence for Conditional Image Generation, [CVPR 2021] Unbalanced Feature Transport for Exemplar-based Image Translation
192 stars 26 forks source link

Problems with replication on the ADE20K dataset #9

Open zyq-lucky opened 2 years ago

zyq-lucky commented 2 years ago

Hello I am trying to reproduce UNITE on the ADE20K dataset, but after training up to about 3 epochs, the learned correspondences start to converge to constant. May I ask if this is as expected? Will it learn the correct correspondence if I continue training? And how many epochs do I need to train?

fnzhan commented 2 years ago

Hi, do you change parameters setting in training scripts?

As the learned correspondences start to converge to constant, you can delete --use_coordconv or decrease --ctx_w to be smaller (e.g., 0.5) in train_ade.sh.