Xiao Zeqi commited on
Commit
7e5523d
·
unverified ·
1 Parent(s): 1a936a7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
- We train on 4 H100 GPUs.
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