Cannot run the loss of loss_contrast_mem.py

[winycg]

Hello, I tried to run contrastive seg with memory bank, but the code seems that it cannot connect to loss_contrast_mem.py and I cannot find a suitable seg_net to conduct in trainer_contrastive.py.

https://github.com/tfzhou/ContrastiveSeg/blob/a2f7392d0c50ecdda36fd41205b77a7bc74f7cb8/segmentor/trainer_contrastive.py#L211

Look forward to your help.

[tfzhou]

I hope that you have tried this script: https://github.com/tfzhou/ContrastiveSeg/blob/main/scripts/cityscapes/hrnet/job_run_h_48_d_4_contrast_mem.sh