ContrastiveSeg
ContrastiveSeg copied to clipboard
Cannot run the loss of loss_contrast_mem.py
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.
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