A question about TEM module loss

Hello, Mr. Lin, very thanks, the code is useful for me.
Sorry to be a bother.
When I try to run it, the TEM testing loss is almost equal from epoch 1 to 20, the module learned nothing.
What maybe cause this question? Is there any advises?
Looking forward to your reply, thank you very much!
