Saving weights and logs of step 5000

events.out.tfevents.1635139153.t1v-n-f6f5b6cc-w-0.172878.0.v2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e3732d9f11aea70b0917879c6f430df1b0ad13183cc04993f5ee6c0c789ecf4e
+size 735759

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8171bd7b4c4b826e10efe581ac427a43898d15d06824d22bfb0be2754aee7d27
+oid sha256:abc32f749ba61e10ea9b7109662d7727283c5ef225d192647f2421c085d3543b
 size 497764120

run.sh

@@ -15,7 +15,7 @@ python run_clm_flax.py \
     --warmup_steps="0" \
     --adam_beta1="0.9" --adam_beta2="0.98" --weight_decay="0.01" \
     --overwrite_output_dir \
-    --num_train_epochs="30" \
+    --num_train_epochs="20" \
     --logging_steps="500" \
     --save_steps="5000" \
     --eval_steps="5000" \