Skip to content

GitLab

Unverified Commit fd980d53 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[doc] add tqdm for link prediction tutorial (#6802)

parent 02f8b6ce
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
notebooks/stochastic_training/link_prediction.ipynb
View file @ fd980d53
...@@ -145,7 +145,7 @@ ...@@ -145,7 +145,7 @@
"datapipe = datapipe.transform(partial(gb.exclude_seed_edges, include_reverse_edges=True))\n", "datapipe = datapipe.transform(partial(gb.exclude_seed_edges, include_reverse_edges=True))\n",
"datapipe = datapipe.fetch_feature(feature, node_feature_keys=[\"feat\"])\n", "datapipe = datapipe.fetch_feature(feature, node_feature_keys=[\"feat\"])\n",
"datapipe = datapipe.copy_to(device)\n", "datapipe = datapipe.copy_to(device)\n",
"train_dataloader = gb.DataLoader(datapipe, num_workers=0)" "train_dataloader = gb.DataLoader(datapipe)"
], ],
"metadata": { "metadata": {
"id": "LZgXGfBvYijJ" "id": "LZgXGfBvYijJ"
...@@ -315,7 +315,7 @@ ...@@ -315,7 +315,7 @@
"\n", "\n",
"logits = []\n", "logits = []\n",
"labels = []\n", "labels = []\n",
"for step, data in enumerate(eval_dataloader):\n", "for step, data in tqdm.tqdm(enumerate(eval_dataloader)):\n",
" # Get node pairs with labels for loss calculation.\n", " # Get node pairs with labels for loss calculation.\n",
" compacted_pairs, label = data.node_pairs_with_labels\n", " compacted_pairs, label = data.node_pairs_with_labels\n",
"\n", "\n",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment