Set training to GPU and path fix in solver file