The evaluation is blocked when passing the CRT module

[rose-jinyang]

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.

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]

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