Restore ckpt on 11 epochs

config.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fff50b26b1037cfdcdd4d054800267d967d2cb9465a28ea39454728bac8387b3
-size 1009
+oid sha256:1ace5aef92f7880ccb5fd0e7c5f65556d6914dbd134fa1672b46a0533225c036
+size 811

events.out.tfevents.1625665214.t1v-n-c632aea2-w-0.13918.3.v2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5b4481e9acbbcd282b5316205121c98a89fbe7e3d07c17bf44652cd855dd4c6d
-size 12440555
+oid sha256:e58f7bb22d1ac3c7de24543921a9b1d6ebb7962199761097333a5277fab91182
+size 10393677

events.out.tfevents.1625765990.t1v-n-c632aea2-w-0.103318.3.v2

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

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:efd5330e23febf3d297de9fdcfaa0b5cb4a06969c37fe4034f5d4ead66bb24c7
-size 280
+oid sha256:57af9a7352c58ac048eb46adf9400c9c13363e06afad52bb1253a186f6625c16
+size 497764120