Solved by KFrank in post #2 
              
              
                Hello Swapnil! 

I don’t really understand what you are trying to do nor what your issue is. 
But I have a couple of comments: 
When you calculate loss and val_loss you use normedWeights in 
loss_function.  But when you calculate running_loss you do not 
use normedWeights.  Also, running_loss is an …