[GitHub] [spark] WeichenXu123 commented on pull request #40695: [SPARK-42994][ML][CONNECT] PyTorch Distributor support Local Mode with GPU

["WeichenXu123 (via GitHub)" <gi...@apache.org>]

WeichenXu123 commented on PR #40695:
URL: https://github.com/apache/spark/pull/40695#issuecomment-1501260234

   We also need to change `TorchDistributor._run_local_training` implementation.
   
   The existing code it executes pytorch code in client side, but in spark connect case, we should execute pytorch code in server side (we can reuse _run_distributed_training code in the case). 


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: reviews-unsubscribe@spark.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org


---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscribe@spark.apache.org
For additional commands, e-mail: reviews-help@spark.apache.org