Add weights

weights/g3/132/ckpts/.gitattributes

@@ -0,0 +1 @@
+last.ckpt filter=lfs diff=lfs merge=lfs -text

weights/g3/132/ckpts/last.ckpt

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

weights/g3/132/config.yaml

@@ -0,0 +1,60 @@
+data_dir: /shared/g-luo/public/geolocation_via_guidebook_grounding
+out_dir: /shared/g-luo/geoguessr/models/im2gps_kb/public
+wandb_dir: geolocation_via_guidebook_grounding
+seed: 132
+model_params:
+  arch: resnet50
+  setting: image_clip_clues
+  name: resnet50_image_clip_clues_fixed-attn-sup
+  weights: ${data_dir}/weights/isn_epoch=014-val_loss=18.4833.ckpt
+  batch_size: 128
+  val_check_interval: 1000
+  optim:
+    params:
+      lr: 0.01
+      momentum: 0.9
+      weight_decay: 0.0001
+      attention_lr: 0.001
+  scheduler:
+    params:
+      gamma: 0.5
+      milestones:
+      - 4
+      - 8
+      - 12
+      - 13
+      - 14
+      - 15
+  loss_weight: ${data_dir}/dataset/loss_weight/countries.json
+  partitionings:
+    ptype: default
+    shortnames:
+    - countries
+    files:
+    - ${data_dir}/dataset/s2_cells/countries.csv
+  skiprows: 0
+  msgpack_train_dir: ${data_dir}/dataset/train/msgpack
+  msgpack_val_dir: ${data_dir}/dataset/val/msgpack
+  train_meta_path: ${data_dir}/dataset/train/train.csv
+  val_meta_path: ${data_dir}/dataset/val/val.csv
+  train_label_mapping: ${data_dir}/dataset/train/label_mapping/countries.json
+  val_label_mapping: ${data_dir}/dataset/val/label_mapping/countries.json
+  key_img_id: id
+  key_img_encoded: image
+  num_workers_per_loader: 6
+  attn_beta: 1.0
+  attn_image_size: 768
+  attn_text_size: 768
+  text_features_file: ${data_dir}/dataset/features/guidebook_roberta_base.pkl
+  image_features_file: ${data_dir}/dataset/features/streetview_clip_rn50x16.pkl
+  attn_layer_norm_type: sigmoid
+  attn_input_norm_type: batch_norm
+  text_labels_file: ${data_dir}/dataset/pseudo_labels/countries.json
+  attn_loss_weight: 50
+  attn_loss_alpha: 0.75
+  attn_loss_type: bce
+trainer_params:
+  gpus: 1
+  max_epochs: 15
+  precision: 16
+  reload_dataloaders_every_n_epochs: 1

weights/isn_epoch=014-val_loss=18.4833.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:96e22842ab9af12ec1e801717cd8f662a723e433eb573ae575c25839f5492964
+size 571821130