File size: 533 Bytes
366bd66
fe8a910
f889858
 
 
 
 
 
366bd66
f889858
 
 
 
 
 
 
 
 
fe8a910
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
---
license: cc-by-4.0
language:
- en
library_name: timm
pipeline_tag: image-classification
tags:
- climate
---
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:

```python
import torch

x = torch.load("moco_geo+tp.pth", map_location="cpu")
x = x["state_dict"]
x = {k.replace("module.encoder_q.", ""): v for k, v in x.items() if "encoder_q" in k and "fc" not in k}
torch.save(x, "resnet50_fmow_rgb_gassl-44b4461b.pth")
```