Fix code style in PyTorch QuickStart (#4821)
* fix two issues * fix two issues
This commit is contained in:
parent
e7c7ab2b8e
commit
04542bc6db
1 changed files with 3 additions and 2 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue