Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+mtg_weights.safetensorsh filter=lfs diff=lfs merge=lfs -text

experiment_cfg.yaml

@@ -0,0 +1,81 @@
+dtype: bfloat16
+batch_size: 8
+lr: 0.001
+weight_decay: 0.0
+epochs: 20
+scheduler_step: 10
+load_ckpt:
+losses:
+- semantic_in
+- semantic_out
+- visual_in
+- visual_out
+data:
+  train:
+    dataset_name: curated
+    dataset_path: 
+    img_size:
+    - 768
+    - 768
+    num_training_samples: -1
+    seed: 42
+    train_val_split: 0.8
+    num_visualization_samples: 15
+    filtering:
+      skewness_thresh: -1
+      lpips_thresh: -1
+      source_target_rel_sz_diff: -1
+      matching_score_thresh: 0.7
+    augmentations:
+      enabled: true
+      horizontal_flip_prob: 0.5
+      color_jitter_prob: 0.5
+      brightness_range:
+      - 0.9
+      - 1.1
+      contrast_range:
+      - 0.8
+      - 1.3
+      saturation_range:
+      - 0.8
+      - 1.3
+    num_samples: -1
+  val:
+    dataset_name: curated
+    dataset_path: 
+    img_size:
+    - 768
+    - 768
+    num_training_samples: -1
+    seed: 42
+    train_val_split: 0.8
+    num_visualization_samples: 15
+    filtering:
+      matching_score_thresh: 0.7
+    augmentations:
+      enabled: false
+    num_samples: -1
+model:
+  name: MindTheGlitchModel
+  masked: true
+  freeze_cleandift: true
+  lr_alpha: 10
+  resnet:
+    num_blocks: 1
+    projection_dim: 384
+    num_norm_groups: 32
+  final_feat_resolution: 48
+  feat_type:
+  - us6
+  - us7
+  - us8
+  - us9
+  - us10
+gpu: '0'
+metrics:
+- vsm
+- oracle
+metric_thresholds:
+- 0.5
+- 0.6
+- 0.7

mtg_weights.safetensorsh

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