Xiao Zeqi commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -68,9 +68,9 @@ To enable cloud logging with [Weights & Biases (wandb)](https://wandb.ai/site),
|
|
| 68 |
|
| 69 |
Download pretrained weights from [Oasis](https://github.com/etched-ai/open-oasis).
|
| 70 |
|
| 71 |
-
|
| 72 |
We observe that gradually increasing task difficulty improves performance. Thus, we adopt a multi-stage training strategy:
|
| 73 |
-
|
| 74 |
```bash
|
| 75 |
sh train_stage_1.sh # Small range, no vertical turning
|
| 76 |
sh train_stage_2.sh # Large range, no vertical turning
|
|
|
|
| 68 |
|
| 69 |
Download pretrained weights from [Oasis](https://github.com/etched-ai/open-oasis).
|
| 70 |
|
| 71 |
+
Training the model on 4 H100 GPUs, it converges after approximately 500K steps.
|
| 72 |
We observe that gradually increasing task difficulty improves performance. Thus, we adopt a multi-stage training strategy:
|
| 73 |
+
,
|
| 74 |
```bash
|
| 75 |
sh train_stage_1.sh # Small range, no vertical turning
|
| 76 |
sh train_stage_2.sh # Large range, no vertical turning
|