Please unblind the output information to verify the result correctly

[lxf]

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

[maik_froebe]

Dear @lxf,

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

Best regards,

Maik