Thanks for the reply.
If I have a batch size of 16, how will a python float (I am assuming that you meant a single number) help?
I want to collect loss over all the batches in each epoch and average them to get/ calculate “batch loss per epoch” after the end of epoch.
Would like to draw your attention, @ptrblck