Fix code style in PyTorch QuickStart (#4821)

* fix two issues

* fix two issues
This commit is contained in:
Nancy 2022-06-14 09:58:43 +08:00 committed by GitHub
parent e7c7ab2b8e
commit 04542bc6db

View file

@ -26,6 +26,7 @@ pip install jep==3.9.0
```python
from bigdl.orca import init_orca_context, stop_orca_context
cluster_mode = "local"
if cluster_mode == "local": # For local machine
init_orca_context(cores=4, memory="10g")
elif cluster_mode == "k8s": # For K8s cluster