Skip to content

Commit

Permalink
Merge branch 'main' into ref-proc-call
Browse files Browse the repository at this point in the history
  • Loading branch information
zhanglei1949 authored May 24, 2024
2 parents c08997c + b027c72 commit cee46ba
Showing 1 changed file with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit cee46ba

Please sign in to comment.