add run.sh
Browse files
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" \
|