Skip to content

Commit

Permalink
Fix the bugs in the ds1000 sample bash script
Browse files Browse the repository at this point in the history
  • Loading branch information
gameofby committed Dec 10, 2024
1 parent 6116c6a commit 0507022
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion docs/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -249,7 +249,8 @@ Below is the command to run evaluation on the full benchmark in insertion mode w

```bash
export TF_FORCE_GPU_ALLOW_GROWTH=true
TF_CPP_MIN_LOG_LEVEL=3 accelerate launch main.py \
export TF_CPP_MIN_LOG_LEVEL=3
accelerate launch main.py \
--model <MODEL_NAME> \
--batch_size <BATCH_SIZE> \
--tasks ds1000-all-insertion \
Expand Down

0 comments on commit 0507022

Please sign in to comment.