AWS EC2 : Training a model using jupyter notebooks hosted on a EC2 instance. Kernel keeps dying

Question

I am running a Ubuntu 18.04 instance with t3a.2xlarge instance type of EC2. My model is not even image data, it is big but not that big. The kernel dies just about a minute or