Skip to content

Commit

Permalink
Merge pull request mlcommons#669 from davidmochen/dlrm_readme
Browse files Browse the repository at this point in the history
Mention gsutil tool in README
  • Loading branch information
davidmochen authored Jul 28, 2020
2 parents 669d0e0 + b64af25 commit 5cd926b
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion v0.7/recommendation/dlrm/tf/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,5 +43,16 @@ python3 [DLRM code directory]/dlrm_main.py \
--vocab_sizes_embed=39884406,39043,17289,7420,20263,3,7120,1543,63,38532951,2953546,403346,10,2208,11938,155,4,976,14,39979771,25641295,39664984,585935,12972,108,36
```

A model checkpoint is available in this public GCS bucket:
A model checkpoint is available in this public GCS bucket:
gs://dmchen-data/dlrm_tf1_data/model_dir_0.

The Criteo dataset is stored in a format compatible with the model in this
public GCS bucket:
gs://dmchen-data/dlrm_tf1_data/criteo.

The model checkpoint and Criteo dataset files can be accessed with the
[gsutil tool](https://cloud.google.com/storage/docs/gsutil).

```sh
gsutil cp -r gs://dmchen-data/dlrm_tf1_data/model_dir_0 [your folder]
```

0 comments on commit 5cd926b

Please sign in to comment.