Is there a calculation code for LV loss?

HUANGLIZI / LViT

[IEEE Transactions on Medical Imaging/TMI] This repo is the official implementation of "LViT: Language meets Vision Transformer in Medical Image Segmentation"

MIT License

283 stars 26 forks source link

Is there a calculation code for LV loss? #27

Closed zhouke0304 closed 11 months ago

zhouke0304 commented 12 months ago

I did not find the calculation code related to LV loss and EPI in the code, may I ask how to train under semi-supervised conditions?

HUANGLIZI commented 11 months ago

Thanks for your message. You can refer to the COSINESIMILARITY function. https://pytorch.org/docs/stable/generated/torch.nn.CosineSimilarity.html