The total loss has not contained the layer-wise loss.
The total loss has not contained the layer-wise loss.