Can you provide the original training code? Which loss function is used to train the network? Thanks!