search

IBM / terratorch

a Python toolkit for fine-tuning Geospatial Foundation Models (GFMs).
Apache License 2.0
147 stars 14 forks source link

Allow `predict_dataset_bands` to be specified at `predict` time. #183

Open CarlosGomes98 opened 4 hours ago

CarlosGomes98 commented 4 hours ago

Is your feature request related to a problem? Please describe. When using the model for inference, predict_dataset_bands must be defined at init time.

Describe the solution you'd like I would like to be able to provide this at predict time, so that each call of predict can specify its own predict_dataset_bands.

CarlosGomes98 commented 4 hours ago

The implementation should be relatively straight forward - whatever we currently do in the init can also be done at predict, with an extra argument accepted by the LightningInferenceModel at that time