# Checkpoints

Place the pretrained model checkpoint here as `pretrained.pt`.

This file contains the `state_dict` of a `Seq2SeqNDVI(hidden_dim=64)` model
pretrained on the full training dataset (2020–2022, all grid cells).

To load:

```python
from ndvi_forecast import Seq2SeqNDVI
import torch

model = Seq2SeqNDVI(hidden_dim=64)
model.load_state_dict(torch.load("checkpoints/pretrained.pt", map_location="cpu"))
model.eval()
```
