Thank you,
I have learnt Keras first. I have seen a progress bar like the following:
[============================================================] 100%, 0 seconds.
It is shown immediately once an epoch starts.
I will use your solution, but I wonder if what I have explained can be done using pytorch.