[Bug] High CPU usage during training。