Hello, I think your magin loss is wrong, according to the paper, after relu, there is a square you forgot