Upload completed Stage 2 checkpoints

README.md

@@ -44,9 +44,11 @@ weights.
 | `CHECKPOINT_LICENSE.md` | `CHECKPOINT_LICENSE.md` | `da3b25759abb` |
 | `checkpoints/stage1_autoencoder_best_eval_recon.pt` | `best_eval_recon.pt` | `6f01ec6d1ded` |
 | `configs/autoencoder_photo10k.yaml` | `autoencoder_photo10k.yaml` | `b1743ae89efc` |
-| `checkpoints/stage2_latent_pretrain_step_0006000.pt` | `step_0006000.pt` | `ab329e28b48d` |
 | `configs/latent_pretrain_photo10k.yaml` | `latent_pretrain_photo10k.yaml` | `85175fa83f41` |
-| `metrics/stage2_step_0006000_metrics.json` | `step_0006000_metrics.json` | `261d5847fb93` |
+| `checkpoints/stage2_latent_pretrain_best_eval_latent.pt` | `best_eval_latent.pt` | `54cfddd7003d` |
+| `checkpoints/stage2_latent_pretrain_step_0050000.pt` | `step_0050000.pt` | `40fc175c70d0` |
+| `metrics/stage2_step_0048000_metrics.json` | `step_0048000_metrics.json` | `4db865fc2dbc` |
+| `metrics/stage2_step_0050000_metrics.json` | `step_0050000_metrics.json` | `52367cdebf25` |
 
 ## Stages