acul3 commited on
Commit
ceffc04
1 Parent(s): e41f65b

add run.sh

Browse files
Files changed (1) hide show
  1. run.sh +1 -1
run.sh CHANGED
@@ -1,5 +1,5 @@
1
  # From https://arxiv.org/pdf/1907.11692.pdf
2
- python3 -c "import jax; print('TPUs', jax.device_count())"
3
  tmux new-session -d -s 0 python3 run_mlm_flax.py \
4
  --output_dir="./outputs" \
5
  --model_type="roberta" \
 
1
  # From https://arxiv.org/pdf/1907.11692.pdf
2
+ ##python3 -c "import jax; print('TPUs', jax.device_count())"
3
  tmux new-session -d -s 0 python3 run_mlm_flax.py \
4
  --output_dir="./outputs" \
5
  --model_type="roberta" \