| license: apache-2.0 | |
| # B200 memory architecture checkpoints | |
| This repository contains checkpoints from the | |
| `streaming_3d_perception_online_memory` project. | |
| | Architecture | Training step | Checkpoint | Configuration | | |
| | --- | ---: | --- | --- | | |
| | GDN-2 | 20,000 | `gdn2/checkpoint_000020000.pt` | `gdn2/config.yaml` | | |
| | LaCT | 20,000 | `lact/checkpoint_000020000.pt` | `lact/config.yaml` | | |
| | Mamba-3 | 20,000 | `mamba3/checkpoint_000020000.pt` | `mamba3/config.yaml` | | |
| | Transformer | 13,000 | `transformer/checkpoint_000013000.pt` | `transformer/config.yaml` | | |
| The runs used four B200 nodes with eight GPUs per node, for 32 B200 GPUs in | |
| total. These are complete training checkpoints and may contain model, | |
| optimizer, scheduler, and random-number-generator state. | |
| See `SHA256SUMS` for checkpoint integrity hashes. | |