Issue: Kernel dies when fitting Pytorch model Linux Mint

Can you try running your script under gdb and report the backtrace?

$ gdb --args python my_script.py
...
Reading symbols from python...done.
(gdb) run
...
(gdb) backtrace
...