I have looked through the code, but I couldn't find how to train the network. Could you please tell me how to train the network?