RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed. Specify retain_graph=True when calling backward the first time

You can check the tutorials on how to train a neural network and what each function is doing.