|
--- |
|
license: apache-2.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") |
|
``` |
|
|
