File size: 815 Bytes
51acccf dd5455a 51acccf dd5455a 99d764d dd5455a 14e1f34 dd5455a 14e1f34 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | ---
library_name: pytorch
datasets:
- marplan6/batgrad
tags:
- battery
- timeseries
- foundation-models
- template
license: mit
---
# batgrad model
A checkpoint for exploring the
[`batgrad`](https://github.com/marplan/batgrad) training and inference workflow.
Download it with:
```sh
uv run scripts/hf_assets.py download --ckpt init_baseline
```
The checkpoint is written to
`outputs/checkpoints/init_baseline.pt`. It contains weights for
the hybrid attention/Mamba-3 model, the complete experiment configuration,
training step, checkpoint format version, and creating `batgrad` Git commit.
Load it with `batgrad.ml.checkpoint.load_checkpoint()`. Mamba-3 checkpoints
require Linux and CUDA. The checkpoint can be evaluated on any compatible
normalized `batgrad` dataset. (~10M parameters, ~4GB VRAM inference) |