Skip to content

Commit

Permalink
Review getting started examples (NVIDIA-Merlin#779)
Browse files Browse the repository at this point in the history
* update 04-Tritone-Inference

* update

* update

* update

* start working on pytorch example

* get HugeCTR to work

* implement hugectr tests

* get hugectr test to pass

* fix PyTorch, attempt to fix TF, edit both

* update

* add tests

* update

* update PyTorch example

* update

* fix description of training with TF

* fix tf test

* update

* update

* fix tensorflow notebooks / test

* update pytorch example track

* update

* move tests to tests/unit

* implement review suggestions

* update

* update

* update

* fix tests

* update

* update

* update

* update
  • Loading branch information
radekosmulski authored Jan 22, 2023
1 parent f776f56 commit 5d9113e
Show file tree
Hide file tree
Showing 10 changed files with 1,333 additions and 1,645 deletions.
8 changes: 4 additions & 4 deletions examples/getting-started-movielens/01-Download-Convert.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
"cells": [
{
"cell_type": "code",
"execution_count": null,
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
Expand Down Expand Up @@ -77,7 +77,7 @@
"outputs": [],
"source": [
"INPUT_DATA_DIR = os.environ.get(\n",
" \"INPUT_DATA_DIR\", os.path.expanduser(\"~/nvt-examples/movielens/data/\")\n",
" \"INPUT_DATA_DIR\", os.path.expanduser(\"/workspace/nvt-examples/movielens/data/\")\n",
")"
]
},
Expand All @@ -97,8 +97,8 @@
"name": "stderr",
"output_type": "stream",
"text": [
"downloading ml-25m.zip: 262MB [00:06, 42.1MB/s] \n",
"unzipping files: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 8/8 [00:04<00:00, 1.74files/s]\n"
"downloading ml-25m.zip: 262MB [01:22, 3.18MB/s] \n",
"unzipping files: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 8/8 [00:04<00:00, 2.00files/s]\n"
]
}
],
Expand Down
591 changes: 510 additions & 81 deletions examples/getting-started-movielens/02-ETL-with-NVTabular.ipynb

Large diffs are not rendered by default.

376 changes: 77 additions & 299 deletions examples/getting-started-movielens/03-Training-with-HugeCTR.ipynb

Large diffs are not rendered by default.

352 changes: 105 additions & 247 deletions examples/getting-started-movielens/03-Training-with-PyTorch.ipynb

Large diffs are not rendered by default.

692 changes: 224 additions & 468 deletions examples/getting-started-movielens/03-Training-with-TF.ipynb

Large diffs are not rendered by default.

Loading

0 comments on commit 5d9113e

Please sign in to comment.