Saving weights and logs of step 40000

events.out.tfevents.1641147502.t1v-n-79f0077b-w-0.611431.0.v2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b7a5c25da1827489921712ade9b762727d6a5e9d398c10fc5005e19fd4002cb4
-size 4437794
+oid sha256:15ad8a4b6d7baac2b7f3538669bea52602fc66b55b62cc10d54374dd8b121070
+size 5928006

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:371aeeb2bf25df207ecf46f54bfc20a97ac0410b5af890ac38f70911270ff3bf
+oid sha256:2f21b5f7d326d1e4529d02b835bcb6111bf3e688c37b47563620eb5cb561a034
 size 498796983