Saving weights and logs of step 10001

events.out.tfevents.1637101724.t1v-n-358ff5d1-w-0.857283.3.v2

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

flax_model.msgpack

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

run_stream_128.sh

@@ -13,7 +13,7 @@
     --warmup_steps="5000" \
     --overwrite_output_dir \
     --cache_dir /mnt/disks/flaxdisk/cache/ \
-    --num_train_steps="180000" \
+    --num_train_steps="450000" \
     --adam_beta1="0.9" \
     --adam_beta2="0.98" \
     --logging_steps="10000" \

run_stream_512.sh

@@ -13,7 +13,7 @@
     --warmup_steps="2000" \
     --overwrite_output_dir \
     --cache_dir /mnt/disks/flaxdisk/cache/ \
-    --num_train_steps="20000" \
+    --num_train_steps="70000" \
     --adam_beta1="0.9" \
     --adam_beta2="0.98" \
     --logging_steps="20000" \