Update README.md
Browse files
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
|
| 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
|