TerraTorch - A model training toolkit for geospatial tasks
copied from cf-post-staging / terratorchThe purpose of this package is to build a flexible fine-tuning framework for Geospatial Foundation Models (GFMs) based on TorchGeo and Lightning which can be employed at different abstraction levels. It supports models from the Prithvi, TerraMind, and Granite series as well as models from TorchGeo and timm.
This library provides