I have had this freeze during a model training session too, when the machine was unattended, so I think it may be GPU-related. I thought for the longest time it was power management related. But maybe that was another freeze bug.