Skip to content

Commit

Permalink
Merge pull request #5 from anandhu-eng/cm_readme_inference_update
Browse files Browse the repository at this point in the history
dlrm modifications - filename and dataset downloads
  • Loading branch information
anandhu-eng authored Aug 12, 2024
2 parents 898ac2a + 8e1eb75 commit a6429f3
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ The benchmark implementation run command will automatically download the validat

### Get Validation Dataset
```
cm run script --tags=get,dataset,criteo,validation -j
cm run script --tags=get,dataset,criteo,_validation -j
```
## Model
The benchmark implementation run command will automatically download the required model and do the necessary conversions. In case you want to only download the official model, you can use the below commands.
Expand All @@ -25,6 +25,6 @@ Get the Official MLPerf DLRM v2 Model

### Pytorch
```
cm run script --tags=get,ml-model,dlrm_v2,_pytorch -j
cm run script --tags=get,ml-model,dlrm,_pytorch -j
```

0 comments on commit a6429f3

Please sign in to comment.