blumenstiel commited on
Commit
a20e470
·
1 Parent(s): ff853ee

Update ReadMe

Browse files
Files changed (1) hide show
  1. README.md +9 -0
README.md CHANGED
@@ -54,6 +54,15 @@ Example Notebooks:
54
  [Landslide Segmentation](https://github.com/NASA-IMPACT/Prithvi-EO-2.0/blob/main/examples/example_landslide4sense.ipynb) [<b><i>>>Try it on Colab<<</i></b>](https://colab.research.google.com/github/NASA-IMPACT/Prithvi-EO-2.0/blob/main/examples/example_landslide4sense.ipynb) (Choose T4 GPU runtime)
55
  [Carbon Flux Prediction (Regression)](https://github.com/NASA-IMPACT/Prithvi-EO-2.0/blob/main/examples/carbon_flux/main_flux_finetune_baselines_trainer.ipynb)
56
 
 
 
 
 
 
 
 
 
 
57
  ### Feedback
58
 
59
  Your feedback is invaluable to us. If you have any feedback about the model, please feel free to share it with us. You can do this by starting a discussion in this HF repository or submitting an issue to [TerraTorch](https://github.com/IBM/terratorch) on GitHub.
 
54
  [Landslide Segmentation](https://github.com/NASA-IMPACT/Prithvi-EO-2.0/blob/main/examples/example_landslide4sense.ipynb) [<b><i>>>Try it on Colab<<</i></b>](https://colab.research.google.com/github/NASA-IMPACT/Prithvi-EO-2.0/blob/main/examples/example_landslide4sense.ipynb) (Choose T4 GPU runtime)
55
  [Carbon Flux Prediction (Regression)](https://github.com/NASA-IMPACT/Prithvi-EO-2.0/blob/main/examples/carbon_flux/main_flux_finetune_baselines_trainer.ipynb)
56
 
57
+ If you plan to use Prithvi in your custom PyTorch pipeline, you can build the backbone with:
58
+ ```python
59
+ from terratorch.registry import BACKBONE_REGISTRY
60
+
61
+ model = BACKBONE_REGISTRY.build("prithvi_eo_v2_tiny_tl", pretrained=True)
62
+ ```
63
+
64
+ Find more information on model usage in our [Prithvi Docs](https://ibm.github.io/terratorch/stable/guide/prithvi_eo/).
65
+
66
  ### Feedback
67
 
68
  Your feedback is invaluable to us. If you have any feedback about the model, please feel free to share it with us. You can do this by starting a discussion in this HF repository or submitting an issue to [TerraTorch](https://github.com/IBM/terratorch) on GitHub.