crosscontrast ep40 | acc=0.9031 | final

crosscontrast/config.json

@@ -5,6 +5,6 @@
   "latent_dim": 256,
   "best_val_accuracy": 0.9030947828292847,
   "epoch": 40,
-  "upload_reason": "new_best",
+  "upload_reason": "final",
   "temperature": 0.05152351036667824
 }

crosscontrast/training_config.json

@@ -0,0 +1,17 @@
+{
+  "n_samples": 500000,
+  "epochs": 40,
+  "batch_size": 4096,
+  "lr": 0.002,
+  "weight_decay": 0.0001,
+  "optimizer": "AdamW",
+  "scheduler": "cosine_with_warmup",
+  "warmup_epochs": 3,
+  "loss": "symmetric_InfoNCE",
+  "initial_temperature": 0.07,
+  "final_temperature": 0.05152351036667824,
+  "amp_dtype": "torch.bfloat16",
+  "best_val_accuracy": 0.9030947828292847,
+  "nearest_text_accuracy": 1.0,
+  "total_training_time_seconds": 12314.070337057114
+}