Aman K
commited on
Commit
•
7e9b25b
1
Parent(s):
208bddf
Saving weights and logs of epoch 1
Browse files
run.sh
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
./run_mlm_flax.py \
|
|
|
|
| 3 |
--output_dir="./" \
|
| 4 |
--model_type="roberta" \
|
| 5 |
--config_name="./" \
|
|
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
./run_mlm_flax.py \
|
| 3 |
+
--model_name_or_path="flax_model.msgpack" \
|
| 4 |
--output_dir="./" \
|
| 5 |
--model_type="roberta" \
|
| 6 |
--config_name="./" \
|