| 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") | |
| ``` |