Skip to content

Commit

Permalink
Update local_test.sh
Browse files Browse the repository at this point in the history
Add more description.
  • Loading branch information
crazydonkey200 authored Mar 1, 2025
1 parent 118b895 commit cbb80a5
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions hero/local_test.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
#!/bin/bash

# Example command for local run.
export JAX_DISABLE_JIT=True; EXP=local_test_1; rm -rf /tmp/${EXP}; python main.py --experiment_config TransformerLMTest --experiment_dir /tmp/${EXP} --verbosity=-1
# Example command for local run.
# Add "export JAX_DISABLE_JIT=True;" to disable `jit` for easier debugging.
# Change "TransformerLMTest" to other experiment config names in `config_lib.py` to run other experiments.
EXP=local_test_1; rm -rf /tmp/${EXP}; python main.py --experiment_config TransformerLMTest --experiment_dir /tmp/${EXP} --verbosity=-1

# Check training curves.
tensorboard --logdir /tmp/${EXP}

0 comments on commit cbb80a5

Please sign in to comment.