isaaccorley commited on
Commit
f889858
1 Parent(s): 366bd66

Upload 2 files

Browse files
Files changed (2) hide show
  1. README.md +16 -0
  2. resnet50_fmow_rgb_gassl-44b4461b.pth +3 -0
README.md CHANGED
@@ -1,3 +1,19 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ language:
4
+ - en
5
+ library_name: timm
6
+ pipeline_tag: image-classification
7
+ tags:
8
+ - climate
9
  ---
10
+ Weights for the MoCo-v2+Geo+TP model from https://github.com/sustainlab-group/geography-aware-ssl, adapted to only extract the encoder using the following code:
11
+
12
+ ```python
13
+ import torch
14
+
15
+ x = torch.load("moco_geo+tp.pth", map_location="cpu")
16
+ x = x["state_dict"]
17
+ x = {k.replace("module.encoder_q.", ""): v for k, v in x.items() if "encoder_q" in k and "fc" not in k}
18
+ torch.save(x, "resnet50_fmow_rgb_gassl-44b4461b.pth")
19
+ ```
resnet50_fmow_rgb_gassl-44b4461b.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da43d987dfaa6e0d481c8633205dc1a16a4296c32492fe1718b4cfd17735eb05
3
+ size 94321093