How do I add something to code to calculate the SSIM?

[GuTeng-ai]

Dear author:
I want to calculate the SSIM on my own dataset in this DAT-S Model, but how do I add something to this current yaml file? And, whether I need to add some code for the train.py or other related part in this project.

[zhengchen1999]

Hello, there's no need to modify the train.py file.
You need to modify the path to your test dataset in the yml file.