Upload Stage 1 and Stage 2 checkpoints

metrics/stage2_step_0004000_metrics.json

@@ -0,0 +1,10 @@
+{
+  "metrics": {
+    "eval/decoded_mse": 0.020275677144527434,
+    "eval/decoded_psnr": 22.950846241182102,
+    "eval/latent_loss": 0.25679622411727904,
+    "eval/latent_mse": 0.5862508678436279,
+    "eval/num_images": 100.0
+  },
+  "step": 4000
+}