Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
nightdessert authored May 13, 2024
1 parent f59125c commit 2d9fb9d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ A Single 80G GPU is enough to detect up to 50K length.
```python
python retrieval_head_detection.py --model_path $path_to_model --s 0 --e 50000
```
We find that only few samples can stablely detect retrieval heads. I if you are in a hurry or no fancy large GPUs avalible, you can just set '--e' to a lower value, e.g.
We find that only few samples can stablely detect some of the strongest retrieval heads. I if you are in a hurry or no fancy large GPUs avalible, you can just set '--e' to a lower value, e.g.
```python
python retrieval_head_detection.py --model_path $path_to_model --s 0 --e 5000
```
Expand Down

0 comments on commit 2d9fb9d

Please sign in to comment.