Saving weights and logs of step 5000

events.out.tfevents.1639508897.t1v-n-ccbf3e94-w-0.1127841.3.v2

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

events.out.tfevents.1639509101.t1v-n-ccbf3e94-w-0.1129530.3.v2

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

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e71edca5819ab8aa0ba904fe14c91587deec98559ac47277f89bdc697555ce60
+oid sha256:d4bfa49f9daaa21db6aaf482ff37c197589cabf8e1b47fb19363e76de72daf02
 size 498796983

run_w5_long.sh

@@ -10,7 +10,7 @@
     --weight_decay="0.01" \
     --per_device_train_batch_size="128" \
     --per_device_eval_batch_size="128" \
-    --learning_rate="3e-4" \
+    --learning_rate="2e-4" \
     --warmup_steps="5000" \
     --overwrite_output_dir \
     --num_train_epochs="250" \