yes it is the loss of train and validation.
Why is the validation loss parallel to the x -axis and training loss so quickly going down?
I want to understand what my model is suffering from?
can you give me an example of using register_hook during a training iteration?