acul3 commited on
Commit
a1ae28e
1 Parent(s): c3f634d

add requirements

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