Why do I execute an epoch in the train section every time I restart the kernel?
You are probably overloading the CPU and/or memory because you are probably doing something wrong in coding!