Skip to content

Commit 83a6149

Browse files
committed
DLRM examples to create prediction using TorchRec library. This shows basic example of how to use TorchRec library quickly locally. (#3043)
Summary: Pull Request resolved: #3043 Rollback Plan: Differential Revision: D75989524
1 parent d639294 commit 83a6149

File tree

3 files changed

+540
-0
lines changed

3 files changed

+540
-0
lines changed

examples/prediction/README.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
# Running torchrec with torchx using Ray scheduler on a Ray cluster
2+
3+
```
4+
pip install --pre torchrec -f https://download.pytorch.org/whl/torchrec/index.html
5+
```
6+
7+
run locally
8+
```
9+
python3 predict_using_torchrec.py
10+
```

examples/prediction/__init__.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
#!/usr/bin/env python3
2+
# Copyright (c) Meta Platforms, Inc. and affiliates.
3+
# All rights reserved.
4+
#
5+
# This source code is licensed under the BSD-style license found in the
6+
# LICENSE file in the root directory of this source tree.
7+
8+
# pyre-strict
9+
10+
11+
def main() -> None:
12+
"""DOC_STRING"""
13+
14+
15+
if __name__ == "__main__":
16+
main()

0 commit comments

Comments
 (0)