Is the process being killed immediately or after a few iterations? You might want to “instrument” your program and seeing if removing some data loading steps or inputs causes the crash to go away to see where the actual problem lies.