fix(learning): Adjust setting for ack cluster (#3842)

alibaba · May 24, 2024 · b027c72 · b027c72
1 parent 11622ec
commit b027c72
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/graphscope/learning/gl_torch_examples/k8s/k8s_launch.py b/python/graphscope/learning/gl_torch_examples/k8s/k8s_launch.py
@@ -27,12 +27,13 @@
 # load the ogbn_arxiv graph as an example.
 sess = gs.session(
     with_dataset=True,
-    k8s_service_type="NodePort",
+    k8s_service_type="LoadBalancer",
+    k8s_namespace="default",  # to guarantee auth to launch the pytorchjobs
     k8s_vineyard_mem="8Gi",
     k8s_engine_mem="8Gi",
     vineyard_shared_mem="8Gi",
     k8s_image_pull_policy="IfNotPresent",
-    k8s_image_tag="0.28.0a20240416",
+    k8s_image_tag="0.28.0a20240511",
     num_workers=params["NUM_SERVER_NODES"],
 )
 g = load_ogbn_arxiv(sess=sess, prefix="/dataset/ogbn_arxiv")