When I get this error it is beacuse I get an Out Of memory (OOM) error, so the training is taking more GBs of RAM/GPU than the available, and then the operative system kills the process. Could this be happening you?