cahya commited on
Commit
3c50ea6
1 Parent(s): f5295fb

updated the wandb variables, and start from scratch

Browse files
Files changed (1) hide show
  1. run_pretraining.sh +2 -3
run_pretraining.sh CHANGED
@@ -1,9 +1,8 @@
1
- export WANDB_ENTITY="cahya"
2
- export WANDB_PROJECT="flax-gpt2-small-indonesian"
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}" \