Problems with replication on the ADE20K dataset

[zyq-lucky]

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]

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.