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
|
```python
|
||||||
from bigdl.orca import init_orca_context, stop_orca_context
|
from bigdl.orca import init_orca_context, stop_orca_context
|
||||||
|
|
||||||
|
cluster_mode = "local"
|
||||||
if cluster_mode == "local": # For local machine
|
if cluster_mode == "local": # For local machine
|
||||||
init_orca_context(cores=4, memory="10g")
|
init_orca_context(cores=4, memory="10g")
|
||||||
elif cluster_mode == "k8s": # For K8s cluster
|
elif cluster_mode == "k8s": # For K8s cluster
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue