isaaccorley commited on
Commit
e7d2ef4
·
verified ·
1 Parent(s): eb2e36c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -2,7 +2,7 @@
2
  license: mit
3
  ---
4
 
5
- The following provides minimal code for loading and exporting the [Tessera geospatial foundation model](https://github.com/ucam-eo/tessera). The original checkpoint file `best_model_fsdp_20250427_084307.pt` hosted on [Google Drive](https://drive.google.com/drive/folders/18RPptbUkCIgUfw1aMdMeOrFML_ZVMszn?usp=sharing) was ~7GB however repackaging only the model weights results in a 337MB checkpoint file `model.pt`. Further, the model is also exported with torch.export to `model_exported_program.pt2` so that the model code itself is not needed to run inference.
6
 
7
  ```python
8
  import torch
 
2
  license: mit
3
  ---
4
 
5
+ The following provides minimal code for loading and exporting the [Tessera geospatial foundation model](https://github.com/ucam-eo/tessera). The original checkpoint file `best_model_fsdp_20250427_084307.pt` hosted on [Google Drive](https://drive.google.com/drive/folders/18RPptbUkCIgUfw1aMdMeOrFML_ZVMszn?usp=sharing) was ~7GB however repackaging only the model weights results in a 350MB checkpoint file `model.pt`. Further, the model is also exported with torch.export to `model_exported_program.pt2` so that the model code itself is not needed to run inference.
6
 
7
  ```python
8
  import torch