Evaluation/test code

[HOMGH]

Hi, thanks for sharing the source codes.
Can we also have the code for evaluation of the pre-trained model? I just can see the main.py file which is for training the model.
Would be appreciated if you can share the test/evaluation code as well.
Thanks.