[CLIP-ViT-B-32-laion2B-s34B-b79K-uni-768dx768d] artifacts head_best

CLIP-ViT-B-32-laion2B-s34B-b79K-uni-768dx768d/head_best.config.json

@@ -0,0 +1,96 @@
+{
+  "device": "cuda",
+  "seed": 42,
+  "precision": "fp32",
+  "hf_clip_id": "laion/CLIP-ViT-B-32-laion2B-s34B-b79K",
+  "freeze_encoder": true,
+  "normalize_output": true,
+  "dataset": "cifar100",
+  "data_root": "./data",
+  "image_size": 224,
+  "clip_mean": [
+    0.48145466,
+    0.4578275,
+    0.40821073
+  ],
+  "clip_std": [
+    0.26862954,
+    0.26130258,
+    0.27577711
+  ],
+  "train_aug": true,
+  "train_crop_scale": [
+    0.9,
+    1.0
+  ],
+  "hflip_p": 0.5,
+  "vocab_repo": "AbstractPhil/geometric-vocab-768d",
+  "vocab_split": "unicode",
+  "vocab_dim_override": 768,
+  "crystal_dims": 768,
+  "role_index": {
+    "anchor": 0,
+    "support": 1,
+    "contrast": 2,
+    "purpose": 3,
+    "observer": 4
+  },
+  "rose_loss_weight": 0.15,
+  "rose_margin": 0.25,
+  "rose_temperature": 1.0,
+  "rose_role_weights": {
+    "anchor": 1.0,
+    "need": -0.5,
+    "relation": 0.75,
+    "purpose": 0.75,
+    "observer": -0.5
+  },
+  "classes": 100,
+  "classes_per_penta": 2,
+  "penta_branches": 50,
+  "pentachora_blocks": 4,
+  "temperature": 0.07,
+  "batch_size": 512,
+  "epochs": 20,
+  "lr": 0.0001,
+  "weight_decay": 0.0001,
+  "label_smoothing": 0.0,
+  "use_cosine_sched": true,
+  "warmup_steps": 0,
+  "max_steps": null,
+  "amp": true,
+  "grad_clip_norm": null,
+  "log_every": 25,
+  "auto_train": false,
+  "num_workers": 2,
+  "pin_memory": true,
+  "persistent_workers": true,
+  "ckpt_dir": "./checkpoints",
+  "run_name": "CLIP-ViT-B-32-laion2B-s34B-b79K-uni-768dx768d",
+  "save_latest": true,
+  "save_best": true,
+  "resume": false,
+  "resume_which": "best",
+  "resume_source": "auto",
+  "hf_enable": true,
+  "hf_repo_id": "AbstractPhil/rose-geoclip",
+  "hf_branch": "main",
+  "hf_private": true,
+  "hf_push_latest": false,
+  "hf_push_best": true,
+  "hf_create": true,
+  "hf_token_env": [
+    "HF_TOKEN",
+    "HUGGINGFACE_HUB_TOKEN"
+  ],
+  "features_enable": true,
+  "features_local_root": "./features_cache",
+  "features_batch_size": 1024,
+  "features_push_to_hub": true,
+  "features_offload_after_prepare": false,
+  "features_repo_id": "AbstractPhil/CLIP-ViT-B-32-laion2B-s34B-b79K_cifar100_extracted_features",
+  "features_local_dir": "./features_cache/CLIP-ViT-B-32-laion2B-s34B-b79K-cifar100",
+  "features_embed_dim": 512,
+  "features_ready": true,
+  "use_precomputed_features": true
+}

CLIP-ViT-B-32-laion2B-s34B-b79K-uni-768dx768d/head_best.metrics.json

@@ -0,0 +1,10 @@
+{
+  "epoch": 1,
+  "train_loss": 2.7008541645431516,
+  "val_loss": 0.9892485931396484,
+  "top1": 0.8,
+  "top3": 0.9327999994277955,
+  "rose_weight": 0.15,
+  "rose_margin": 0.25,
+  "rose_temperature": 1.0
+}