updated the wandb variables, and start from scratch
Browse files- run_pretraining.sh +2 -3
run_pretraining.sh
CHANGED
|
@@ -1,9 +1,8 @@
|
|
| 1 |
-
export WANDB_ENTITY="
|
| 2 |
-
export WANDB_PROJECT="flax-gpt2-
|
| 3 |
export WANDB_LOG_MODEL="true"
|
| 4 |
|
| 5 |
./run_clm_flax.py \
|
| 6 |
-
--model_name_or_path="flax_model.msgpack" \
|
| 7 |
--output_dir="${MODEL_DIR}" \
|
| 8 |
--model_type="gpt2" \
|
| 9 |
--config_name="${MODEL_DIR}" \
|
|
|
|
| 1 |
+
export WANDB_ENTITY="wandb"
|
| 2 |
+
export WANDB_PROJECT="ihf-flax-gpt2-indonesian"
|
| 3 |
export WANDB_LOG_MODEL="true"
|
| 4 |
|
| 5 |
./run_clm_flax.py \
|
|
|
|
| 6 |
--output_dir="${MODEL_DIR}" \
|
| 7 |
--model_type="gpt2" \
|
| 8 |
--config_name="${MODEL_DIR}" \
|