mhamilton723 / STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences
MIT License
712 stars 143 forks source link

The evaluation is blocked when passing the CRT module #37

Open rose-jinyang opened 2 years ago

rose-jinyang commented 2 years ago

Hello How are you? Thanks for contributing to this project. I am going to run the evaluation script with the pre-trained model. However, when passing the CRT module, the evaluation process is blocked. So I debugged it and found likely problem. In your UnNormalize class implementation, the part cloning an image tensor is blocked.

image

The problem is that this part should run in a child thread. it seems that the CRT module should run in a main thread. I tried to fix this issue but did NOT succeed yet. Any solution?

Supgb commented 2 years ago

https://github.com/mhamilton723/STEGO/issues/16#issuecomment-1168772107 It should help.