Please unblind the output information to verify the result correctly

I fixed the issue and ran on both validation and test dataset. Could you please verify if the last test

Dear @lxf,

Thanks for reaching out, everything looks good, I write a direct message to you for more details.

Best regards,