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.

ContrastiveSeg/segmentor/trainer_contrastive.py

Line 211 in a2f7392

if self.with_memory is True:

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