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]

#16 (comment)
It should help.