Skip to content

Commit 479920a

Browse files
author
JacoCheung
committed
Move root ReadMe env setting up to training
1 parent 7f8c695 commit 479920a

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

examples/hstu/training/README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,20 +3,22 @@ We have supported both retrieval and ranking model whose backbones are HSTU laye
33
## Environment Setup
44
### Start from dockerfile
55

6-
We provide [dockerfile](./docker/Dockerfile) for users to build environment.
6+
We provide [dockerfile](../../../docker/Dockerfile) for users to build environment.
77
```
8+
git clone https://github.com/NVIDIA/recsys-examples.git && cd recsys-examples
89
docker build -f docker/Dockerfile --platform linux/amd64 -t recsys-examples:latest .
910
```
1011
If you want to build image for Grace, you can use
1112
```
13+
git clone https://github.com/NVIDIA/recsys-examples.git && cd recsys-examples
1214
docker build -f docker/Dockerfile --platform linux/arm64 -t recsys-examples:latest .
1315
```
1416
You can also set your own base image with args `--build-arg <BASE_IMAGE>`.
1517

1618
### Start from source file
1719
Before running examples, build and install libs under corelib following instruction in documentation:
18-
- [HSTU attention documentation](./corelib/hstu/README.md)
19-
- [Dynamic Embeddings documentation](./corelib/dynamicemb/README.md)
20+
- [HSTU attention documentation](.../../../corelib/hstu/README.md)
21+
- [Dynamic Embeddings documentation](.../../../corelib/dynamicemb/README.md)
2022

2123
On top of those two core libs, Megatron-Core along with other libs are required. You can install them via pypi package:
2224

0 commit comments

Comments
 (0)