Consolidate full 1.2 training documentation and source archive into one PDF
Browse files- .gitattributes +1 -0
- README.md +5 -3
- docs/TrackFormer_1.2_Training_and_Reproducibility.pdf +3 -0
- docs/trackformer_1_2_training_data.md +3 -60
- training/trackformer_1_2/README.md +5 -59
- training/trackformer_1_2/SOURCE_MANIFEST.json +16 -5
- training/trackformer_1_2/config.json +5 -1
- training/trackformer_1_2/records/selected-checkpoint-training.json +63 -0
.gitattributes
CHANGED
|
@@ -216,3 +216,4 @@ training/trackformer_1_2/data/pressure_hpa.npy filter=lfs diff=lfs merge=lfs -te
|
|
| 216 |
training/trackformer_1_2/data/geography.npz filter=lfs diff=lfs merge=lfs -text
|
| 217 |
training/trackformer_1_2/checkpoints/v168_epoch_003.pt filter=lfs diff=lfs merge=lfs -text
|
| 218 |
training/trackformer_1_2/checkpoints/v173_epoch_004.pt filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 216 |
training/trackformer_1_2/data/geography.npz filter=lfs diff=lfs merge=lfs -text
|
| 217 |
training/trackformer_1_2/checkpoints/v168_epoch_003.pt filter=lfs diff=lfs merge=lfs -text
|
| 218 |
training/trackformer_1_2/checkpoints/v173_epoch_004.pt filter=lfs diff=lfs merge=lfs -text
|
| 219 |
+
docs/TrackFormer_1.2_Training_and_Reproducibility.pdf filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -150,11 +150,13 @@ Routes and central pressures are averaged separately. Physical member pressure f
|
|
| 150 |
|
| 151 |
Code and project weights are **MIT licensed**, including modification, redistribution and commercial use. Source, training arrays and checkpoints are public without account approval or a download gate. [Openness inventory and remaining provenance gaps](OPENNESS.md) · [Contributing](CONTRIBUTING.md) · [Software citation](CITATION.cff). Upstream weather data retain their own terms and attribution.
|
| 152 |
|
| 153 |
-
##
|
| 154 |
|
| 155 |
-
|
| 156 |
|
| 157 |
-
|
|
|
|
|
|
|
| 158 |
|
| 159 |
## Get started
|
| 160 |
|
|
|
|
| 150 |
|
| 151 |
Code and project weights are **MIT licensed**, including modification, redistribution and commercial use. Source, training arrays and checkpoints are public without account approval or a download gate. [Openness inventory and remaining provenance gaps](OPENNESS.md) · [Contributing](CONTRIBUTING.md) · [Software citation](CITATION.cff). Upstream weather data retain their own terms and attribution.
|
| 152 |
|
| 153 |
+
## Complete training manual (PDF)
|
| 154 |
|
| 155 |
+
**[Download the single 106-page training and reproducibility PDF](docs/TrackFormer_1.2_Training_and_Reproducibility.pdf).** It contains data sources and processing, whole-storm partitions, normalization, tensor construction, architecture, initialization, all active losses, sampling, batches, truncated backpropagation, optimizer settings, curriculum, validation, checkpoint/resume behavior, recorded history, reproduction commands and limitations. A complete 40-file source/provenance appendix and an embedded exact-source ZIP are included; no separate Markdown pages are needed to read the method.
|
| 156 |
|
| 157 |
+
The checkpoint audit distinguishes facts from the archived recipe: the selected epoch-4 optimizer records **4,384 updates**, compatible with effective batch **16** under the published four-epoch sampler; the archived notebook recipe uses batch **10** and would produce 7,016 updates. Original argv/microbatch/worker settings were not recorded. The PDF explains that discrepancy, the epoch-2 minimum validation loss versus selected epoch 4, and the remaining raw-provenance/runtime gaps.
|
| 158 |
+
|
| 159 |
+
[Training source and asset index](training/trackformer_1_2/) · [Public exact corpus and optimizer checkpoints](https://huggingface.co/euler314/typhoon-predict/tree/main/training/trackformer_1_2). Project code/weights remain MIT; weather data retain upstream terms. This documentation publication does not rerun or modify any trainer.
|
| 160 |
|
| 161 |
## Get started
|
| 162 |
|
docs/TrackFormer_1.2_Training_and_Reproducibility.pdf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2ff4dfe29509fa62e5cffd91e4046d5ca2d436e606cd8ea0c14f15df33adf7b1
|
| 3 |
+
size 499229
|
docs/trackformer_1_2_training_data.md
CHANGED
|
@@ -1,62 +1,5 @@
|
|
| 1 |
-
#
|
| 2 |
|
| 3 |
-
The
|
| 4 |
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
| Component | Dataset | Variables and sampling |
|
| 8 |
-
| --- | --- | --- |
|
| 9 |
-
| Storm observations / supervised labels | [NOAA IBTrACS](https://www.ncei.noaa.gov/products/international-best-track-archive)-derived `track_windows_v13.npz` | Position, prior motion, available maximum wind and central pressure, exact storm/time identities and label masks. Only eligible Western Pacific-domain windows enter this release. |
|
| 10 |
-
| Basin weather | [NOAA PSL NCEP/NCAR Reanalysis 1](https://psl.noaa.gov/data/gridded/data.ncep.reanalysis.html), `basin_all_int8.npz` and the exact-time SLP atlas | MSLP, 500 hPa geopotential height, zonal/meridional winds at 850/500/200 hPa. Six-hour analyses on 100–180°E, 0–60°N; 2.5° grid, 25 × 33 cells. |
|
| 11 |
-
| Native regional pressure | [ARCO-ERA5](https://github.com/google-research/arco-era5), `pressure_hpa.npy` with its frozen `plan.json` | Mean sea-level pressure on 0.25° fixed issue-centred 121 × 121 patches. 1,000 original windows across all three partitions, not native detail for every coarse window. |
|
| 12 |
-
| Static geography | Same ARCO-ERA5 source, `geography.npz` | Land fraction and surface geopotential / 9.80665 to obtain elevation in metres; latitude/longitude encodings. Static context is not an additional temporal fitting period. |
|
| 13 |
-
|
| 14 |
-
The eight weather channels are **MSLP, HGT500, U850, V850, U500, V500, U200, V200**. No satellite cloud imagery, humidity/vapour, SST or ocean-heat-content channel is present in this released input contract. Such features in other experiments must not be attributed to 1.2.
|
| 15 |
-
|
| 16 |
-
The eastward/northward **u/v pairs encode wind direction and speed together**, not direction alone. These environmental winds evolve during the model rollout; their presence is not a separate claim of validated monsoon or surface-wind skill. The [Chinese research explanation](research_report_zh.md) uses these same source variables and partition counts.
|
| 17 |
-
|
| 18 |
-
The selected dataset manifest identifies the **derived IBTrACS archive by SHA-256**, not by an independently recorded raw IBTrACS release/download ID. A newer downloaded IBTrACS file elsewhere is not evidence of which raw snapshot built these frozen windows. Available wind labels and the learned auxiliary head do not establish a harmonized agency averaging period or a validated surface-wind/radius forecast.
|
| 19 |
-
|
| 20 |
-
## Temporal partitions
|
| 21 |
-
|
| 22 |
-
| Role | Whole-storm years | Eligible windows / storms | Native-pressure windows | First → last eligible UTC issue |
|
| 23 |
-
| --- | --- | --- | ---: | --- |
|
| 24 |
-
| Fitting and normalization | 2000–2021 | 13,949 / 611 | 800 | 2000-05-05 18:00 → 2021-12-20 12:00 |
|
| 25 |
-
| Validation / checkpoint selection | 2022–2023 | 1,041 / 46 | 100 | 2022-03-31 00:00 → 2023-12-17 18:00 |
|
| 26 |
-
| Original test / evaluation | 2024–2025 | 1,195 / 56 | 100 | 2024-05-22 00:00 → 2025-12-02 12:00 |
|
| 27 |
-
|
| 28 |
-
These are loader-eligible window counts, not independent storms or the number of gradient updates. Native-pressure windows are a **subset** of the corresponding eligible windows, not additional cases. Storm membership uses each storm's first year in the frozen archive. Windows whose −48 h history or +120 h targets cross a partition boundary are excluded. Weather normalization uses the 2000–2021 basin frames only.
|
| 29 |
-
|
| 30 |
-
Every forecast history comprises nine consecutive six-hour analyses at **−48, −42, …, 0 h**. Supervised fields and storm labels at **+6, +12, …, +120 h** are targets only; they cannot be inference inputs. Missing native detail and missing intensity labels retain their masks. Validation is not gradient fitting, but it does inform checkpoint selection. Previously inspected original tests and showcases are development evidence, not a new unused holdout.
|
| 31 |
-
|
| 32 |
-
The two-day span is the input format chosen for this implementation. No matched 24/48/72-hour experiment establishes it as the best history length. Likewise, the evaluation's 50 input-perturbed forecasts share one set of weights; the count is not a training-data multiplier or proof of a calibrated ensemble.
|
| 33 |
-
|
| 34 |
-
## Newer data do not change the fitting cutoff
|
| 35 |
-
|
| 36 |
-
The checksum-pinned caches contain records beyond the fitting years:
|
| 37 |
-
|
| 38 |
-
- Derived storm-window archive: **1980-01-03 00 UTC to 2026-07-13 00 UTC**.
|
| 39 |
-
- Coarse NCEP weather archive: **1980-01-01 00 UTC to 2026-03-17 18 UTC**.
|
| 40 |
-
- Original native-pressure plan: fitting, validation and test windows in **2000–2025**, separated as above.
|
| 41 |
-
|
| 42 |
-
Those are **source-cache endpoints**, not years all used to fit 1.2. The 2026 live/backfill pipeline uses the unchanged released model with later causal inputs; GFS transfer is experimental, and generating a new forecast does not retrain the weights. Retrospective reanalysis availability is not proof that the same inputs were available operationally at each historical issue time.
|
| 43 |
-
|
| 44 |
-
## Verified identity and reproduction
|
| 45 |
-
|
| 46 |
-
The [training-data receipt](../evaluation/training_data/trackformer_1_2_provenance.json) rechecks all six derived dataset file hashes, the released dataset-manifest hash, exact split counts, storm separation and native-patch membership. It records the original cache endpoints and exact earliest histories/latest targets. This is not a new raw-source-download audit.
|
| 47 |
-
|
| 48 |
-
| Frozen identity | SHA-256 |
|
| 49 |
-
| --- | --- |
|
| 50 |
-
| Dataset manifest | `04fd0d30620137aad20173dcd59e22da2a6c20fae2f17de593e91a59517459c1` |
|
| 51 |
-
| Selected training checkpoint | `f194a23d3f91ea76ad776dfad942fabd669eeae8b3fd665815463095367e9ee0` |
|
| 52 |
-
| Inference-only weights | `db49f36e85a3766defc4c172746897a1f783705d1ce8e6f9dfb8e87ae1d902cb` |
|
| 53 |
-
|
| 54 |
-
With the original frozen dataset available, reproduce the small receipt without inference, downloads or refreshing the training cache:
|
| 55 |
-
|
| 56 |
-
```bash
|
| 57 |
-
python release_tools/audit_training_data.py --dataset-root /Volumes/D/typhoon_predict/data/v164_reuse --output /Volumes/D/typhoon_predict/remote_typhoon_predict_repo/evaluation/training_data/trackformer_1_2_provenance.json
|
| 58 |
-
```
|
| 59 |
-
|
| 60 |
-
The exact derived arrays, original manifest, warm start and selected full training checkpoint are now publicly available in [the open training package](../training/trackformer_1_2/README.md). Original source bytes and frozen model weights are unchanged. The additive publication does not run training or modify any active trainer.
|
| 61 |
-
|
| 62 |
-
The source archive includes historical ERA5 preparation utilities; its small full-field pilot is not the released six-file corpus. Use the published exact corpus to reproduce the released data boundary. No missing raw IBTrACS provenance has been invented.
|
|
|
|
| 1 |
+
# TrackFormer 1.2 training data and process
|
| 2 |
|
| 3 |
+
The complete explanation is consolidated in **[one 106-page PDF](TrackFormer_1.2_Training_and_Reproducibility.pdf)**, rather than split across Markdown pages. It includes sources, exact corpus, train/validation/test partitions, eligibility, normalization, all input/target tensors and masks, training stages, active losses, optimizer and checkpoint details, reproduction commands, full original text-source appendices and provenance limits.
|
| 4 |
|
| 5 |
+
The original training-data audit remains machine-readable at [evaluation/training_data/trackformer_1_2_provenance.json](../evaluation/training_data/trackformer_1_2_provenance.json). Exact public assets and source are indexed at [training/trackformer_1_2/](../training/trackformer_1_2/). The PDF reproduces the full earlier data explanation and audit, including the unrecorded original raw IBTrACS snapshot ID; it does not invent missing provenance.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
training/trackformer_1_2/README.md
CHANGED
|
@@ -1,63 +1,9 @@
|
|
| 1 |
-
# TrackFormer 1.2
|
| 2 |
|
| 3 |
-
|
| 4 |
|
| 5 |
-
|
| 6 |
|
| 7 |
-
|
| 8 |
-
- `source/data_preparation/`: original historical ERA5 download/plan utilities and dataset contracts. The small full-field pilot described there is **not** the six-file training corpus.
|
| 9 |
-
- `assets.json`: exact sizes and SHA-256 values for all six original derived archives, the original manifest, warm-start checkpoint and selected full training checkpoint.
|
| 10 |
-
- `data/`: those exact public arrays, including original split plan, labels/masks and static geography. Binary files are hosted on Hugging Face at the identical path, not committed into GitHub source history.
|
| 11 |
-
- `checkpoints/v168_epoch_003.pt`: the original, mandatory warm start, previously split across Drive files.
|
| 12 |
-
- `checkpoints/v173_epoch_004.pt`: the selected full training checkpoint, with model, optimizer, normalization, four-epoch history and Python/NumPy/Torch CPU RNG states.
|
| 13 |
-
- `config.json`, `checkpoint-history.json`, `SOURCE_MANIFEST.json`: archived recipe, exact saved loss history and source hashes.
|
| 14 |
-
- `train.py`: a path-portability wrapper; it calls the unmodified original trainer with explicit paths and settings. No module import starts training.
|
| 15 |
-
- Model export, physical pressure-map rendering and benchmark tools remain in the repository's `release_tools/` and `models/trackformer_1_2_field/`.
|
| 16 |
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
## Download and verify
|
| 20 |
-
|
| 21 |
-
From the repository root, after obtaining this source folder:
|
| 22 |
-
|
| 23 |
-
```bash
|
| 24 |
-
python training/trackformer_1_2/download_assets.py
|
| 25 |
-
python training/trackformer_1_2/download_assets.py --verify-only
|
| 26 |
-
python -m pip install -r training/trackformer_1_2/requirements.txt
|
| 27 |
-
```
|
| 28 |
-
|
| 29 |
-
The downloader uses only Python's standard library. It publishes each local file only after exact size/SHA verification and does not overwrite a differing existing file. Use `--revision <immutable-HF-commit>` from `PUBLICATION.md` to pin the upload; even the default `main` URL must match the fixed hashes. For a writable data/output directory on an external disk, copy the package there or set `--destination`; then pass its `data` and `checkpoints` paths to the launcher.
|
| 30 |
-
|
| 31 |
-
## Preflight, then explicitly start training
|
| 32 |
-
|
| 33 |
-
```bash
|
| 34 |
-
python training/trackformer_1_2/train.py --device mps --output runs/retrain-1.2 --preflight
|
| 35 |
-
python training/trackformer_1_2/train.py --device mps --output runs/retrain-1.2
|
| 36 |
-
```
|
| 37 |
-
|
| 38 |
-
Use `--device cuda` for NVIDIA or `--device cpu` for CPU. This wrapper explicitly sets the archived notebook/design recipe: effective batch 10, microbatch 4, workers 3, four epochs; pass flags to change execution settings deliberately. It uses FP32 even though the historical notebook's compatibility AMP positional flag was 1. The unchanged original entry point is also available:
|
| 39 |
-
|
| 40 |
-
```bash
|
| 41 |
-
python training/trackformer_1_2/source/v173/v173_pressure_train.py 10 3 0 \
|
| 42 |
-
--microbatch 4 --epochs 4 --device mps \
|
| 43 |
-
--manifest training/trackformer_1_2/data/manifest.json \
|
| 44 |
-
--warm-start training/trackformer_1_2/checkpoints/v168_epoch_003.pt \
|
| 45 |
-
--output runs/retrain-1.2
|
| 46 |
-
```
|
| 47 |
-
|
| 48 |
-
The original loader hashes all arrays, reconstructs a writable basin mmap beside the dataset, recomputes train-only normalization and enforces counts **13,949 / 1,041 / 1,195** with native-pressure subsets **800 / 100 / 100**. Allow space for that extra approximately 390 MB mmap and approximately 260 MB checkpoint files. Do not point preflight/training at an active trainer's data/output directories. The trainer uses a per-output lock; the same output resumes its `last.pt`. SIGINT/SIGTERM request a checkpoint at an optimizer boundary. Training horizons are 4/8/12/20 six-hour steps; validation always uses 20 steps. Four-step truncated backpropagation, gradient clipping, AdamW settings and all losses are in the original source.
|
| 49 |
-
|
| 50 |
-
## Provenance and honest reproduction limits
|
| 51 |
-
|
| 52 |
-
- Dataset manifest: `04fd0d30620137aad20173dcd59e22da2a6c20fae2f17de593e91a59517459c1`.
|
| 53 |
-
- Warm start: `99ed776097e05ad6df03375465d9618741bed63a4ee57724d3ba98e295039d30`.
|
| 54 |
-
- Selected full checkpoint: `f194a23d3f91ea76ad776dfad942fabd669eeae8b3fd665815463095367e9ee0`.
|
| 55 |
-
- Inference-only weights: `db49f36e85a3766defc4c172746897a1f783705d1ce8e6f9dfb8e87ae1d902cb`.
|
| 56 |
-
|
| 57 |
-
The selected full checkpoint is epoch **4**, used by the report; **it is not the smallest saved aggregate validation loss** (epoch 2 is 0.6939101; epoch 4 is 0.7283264). A fresh run's `best.pt` follows the original minimum-validation criterion and need not be the report's epoch-4 checkpoint. Do not conflate these choices. The supplied warm start makes the actual v173 training path runnable without reconstructing every earlier experimental model. Historical v168 optimization logs and complete original runtime package versions/accelerator RNG states are not archived here. The recipe is preserved from the original notebook/design, not a newly recovered complete original process argv. Bitwise end-to-end retraining across devices has **not** been demonstrated.
|
| 58 |
-
|
| 59 |
-
The exact derived data are published; the raw IBTrACS release/download ID was not recorded in the frozen manifest. Current upstream downloads may differ and cannot be claimed byte-identical. Fitting uses 2000–2021, validation 2022–2023, original test 2024–2025, with complete storms separated. Later cache records are not fitting years. All future fields/positions are labels only. Repeatedly inspected tests/showcases are development evidence, not a fresh unused holdout. This publication does not rerun training, change released weights or modify active rainfall training.
|
| 60 |
-
|
| 61 |
-
## Licenses and attribution
|
| 62 |
-
|
| 63 |
-
Project source and project checkpoints retain the repository's MIT license. Upstream weather/track data retain their own terms and attributions, not the MIT software license; see `THIRD_PARTY_NOTICES.md`. Downloaded full checkpoints use Python pickle containers: load these identified, checksum-verified project artifacts only.
|
|
|
|
| 1 |
+
# TrackFormer 1.2 training
|
| 2 |
|
| 3 |
+
**[Read / download the complete training manual (106-page PDF)](../../docs/TrackFormer_1.2_Training_and_Reproducibility.pdf).**
|
| 4 |
|
| 5 |
+
The single PDF contains the entire training method and data process, active losses, settings, checkpoint history, reproduction commands, limitations and all 40 text-source/provenance files. Its embedded ZIP preserves their exact bytes. You do not need to open separate Markdown documents to read the method. The original source files remain executable and inspectable here.
|
| 6 |
|
| 7 |
+
Large assets are public at the same [Hugging Face training path](https://huggingface.co/euler314/typhoon-predict/tree/main/training/trackformer_1_2); `assets.json` and `download_assets.py` verify their hashes without tokens or Drive permissions. Code and weights use MIT; upstream data attribution remains in `THIRD_PARTY_NOTICES.md` and the PDF.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
+
Important: the archived notebook/config recipe uses batch 10, while the selected full checkpoint records 4,384 updates, compatible with batch 16 for the published four-epoch sampler. The PDF documents the evidence and provides both commands without claiming recovery of the original argv. See `records/selected-checkpoint-training.json` for the machine-readable inspection. The report selects epoch 4; the saved validation-loss minimum is epoch 2.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
training/trackformer_1_2/SOURCE_MANIFEST.json
CHANGED
|
@@ -20,8 +20,8 @@
|
|
| 20 |
},
|
| 21 |
{
|
| 22 |
"path": "training/trackformer_1_2/README.md",
|
| 23 |
-
"sha256": "
|
| 24 |
-
"bytes":
|
| 25 |
},
|
| 26 |
{
|
| 27 |
"path": "training/trackformer_1_2/THIRD_PARTY_NOTICES.md",
|
|
@@ -40,8 +40,8 @@
|
|
| 40 |
},
|
| 41 |
{
|
| 42 |
"path": "training/trackformer_1_2/config.json",
|
| 43 |
-
"sha256": "
|
| 44 |
-
"bytes":
|
| 45 |
},
|
| 46 |
{
|
| 47 |
"path": "training/trackformer_1_2/download_assets.py",
|
|
@@ -132,6 +132,11 @@
|
|
| 132 |
"path": "training/trackformer_1_2/train.py",
|
| 133 |
"sha256": "4502a87bfafb11667015070c6fad37b723b4a53723789febda075fba1c7770e8",
|
| 134 |
"bytes": 1436
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 135 |
}
|
| 136 |
],
|
| 137 |
"assets": [
|
|
@@ -189,5 +194,11 @@
|
|
| 189 |
"bytes": 257969971,
|
| 190 |
"group": "checkpoints"
|
| 191 |
}
|
| 192 |
-
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 193 |
}
|
|
|
|
| 20 |
},
|
| 21 |
{
|
| 22 |
"path": "training/trackformer_1_2/README.md",
|
| 23 |
+
"sha256": "97ac00bff15891218956a4743ae0d12cddad56106683dbc3c14b4922a463023c",
|
| 24 |
+
"bytes": 1328
|
| 25 |
},
|
| 26 |
{
|
| 27 |
"path": "training/trackformer_1_2/THIRD_PARTY_NOTICES.md",
|
|
|
|
| 40 |
},
|
| 41 |
{
|
| 42 |
"path": "training/trackformer_1_2/config.json",
|
| 43 |
+
"sha256": "6dddaba9c17720eedb6ab8ad2564c6da926ff37ba490257a585facb59d477e4e",
|
| 44 |
+
"bytes": 1064
|
| 45 |
},
|
| 46 |
{
|
| 47 |
"path": "training/trackformer_1_2/download_assets.py",
|
|
|
|
| 132 |
"path": "training/trackformer_1_2/train.py",
|
| 133 |
"sha256": "4502a87bfafb11667015070c6fad37b723b4a53723789febda075fba1c7770e8",
|
| 134 |
"bytes": 1436
|
| 135 |
+
},
|
| 136 |
+
{
|
| 137 |
+
"path": "training/trackformer_1_2/records/selected-checkpoint-training.json",
|
| 138 |
+
"sha256": "28a8b86dfd66cc718d7cb0b165af885cbee3f39627fdb9cda0be91c021260180",
|
| 139 |
+
"bytes": 1537
|
| 140 |
}
|
| 141 |
],
|
| 142 |
"assets": [
|
|
|
|
| 194 |
"bytes": 257969971,
|
| 195 |
"group": "checkpoints"
|
| 196 |
}
|
| 197 |
+
],
|
| 198 |
+
"documentation_pdf": {
|
| 199 |
+
"path": "docs/TrackFormer_1.2_Training_and_Reproducibility.pdf",
|
| 200 |
+
"sha256": "2ff4dfe29509fa62e5cffd91e4046d5ca2d436e606cd8ea0c14f15df33adf7b1",
|
| 201 |
+
"pages": 106,
|
| 202 |
+
"source_snapshot_huggingface_revision": "1f4ed985a05589076d37f170cc8053e26cec9fba"
|
| 203 |
+
}
|
| 204 |
}
|
training/trackformer_1_2/config.json
CHANGED
|
@@ -25,5 +25,9 @@
|
|
| 25 |
"selected_epoch": 4,
|
| 26 |
"minimum_saved_validation_loss_epoch": 2,
|
| 27 |
"warm_start_sha256": "99ed776097e05ad6df03375465d9618741bed63a4ee57724d3ba98e295039d30",
|
| 28 |
-
"selected_checkpoint_sha256": "f194a23d3f91ea76ad776dfad942fabd669eeae8b3fd665815463095367e9ee0"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
}
|
|
|
|
| 25 |
"selected_epoch": 4,
|
| 26 |
"minimum_saved_validation_loss_epoch": 2,
|
| 27 |
"warm_start_sha256": "99ed776097e05ad6df03375465d9618741bed63a4ee57724d3ba98e295039d30",
|
| 28 |
+
"selected_checkpoint_sha256": "f194a23d3f91ea76ad776dfad942fabd669eeae8b3fd665815463095367e9ee0",
|
| 29 |
+
"recorded_selected_optimizer_steps": 4384,
|
| 30 |
+
"checkpoint_compatible_inferred_effective_batch": 16,
|
| 31 |
+
"archived_recipe_batch_is_not_verified_original_argv": true,
|
| 32 |
+
"documentation_pdf": "docs/TrackFormer_1.2_Training_and_Reproducibility.pdf"
|
| 33 |
}
|
training/trackformer_1_2/records/selected-checkpoint-training.json
ADDED
|
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"source_checkpoint_sha256": "f194a23d3f91ea76ad776dfad942fabd669eeae8b3fd665815463095367e9ee0",
|
| 3 |
+
"version": "1.2.73",
|
| 4 |
+
"architecture": "v173-moving-core-multiscale-attention",
|
| 5 |
+
"completed_epoch": 4,
|
| 6 |
+
"history": [
|
| 7 |
+
{
|
| 8 |
+
"epoch": 1,
|
| 9 |
+
"horizon": 4,
|
| 10 |
+
"train": 0.1411035786937901,
|
| 11 |
+
"validation": 0.7017406511544373
|
| 12 |
+
},
|
| 13 |
+
{
|
| 14 |
+
"epoch": 2,
|
| 15 |
+
"horizon": 8,
|
| 16 |
+
"train": 0.18676476989154306,
|
| 17 |
+
"validation": 0.6939100875934489
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 3,
|
| 21 |
+
"horizon": 12,
|
| 22 |
+
"train": 0.23248198316925273,
|
| 23 |
+
"validation": 0.7200751373592149
|
| 24 |
+
},
|
| 25 |
+
{
|
| 26 |
+
"epoch": 4,
|
| 27 |
+
"horizon": 20,
|
| 28 |
+
"train": 0.3557772162892601,
|
| 29 |
+
"validation": 0.7283263609942383
|
| 30 |
+
}
|
| 31 |
+
],
|
| 32 |
+
"saved_best": 0.6939100875934489,
|
| 33 |
+
"progress": null,
|
| 34 |
+
"optimizer_step_values": [
|
| 35 |
+
4384
|
| 36 |
+
],
|
| 37 |
+
"optimizer_parameter_states": 375,
|
| 38 |
+
"optimizer_groups": [
|
| 39 |
+
{
|
| 40 |
+
"lr": 3e-05,
|
| 41 |
+
"betas": [
|
| 42 |
+
0.9,
|
| 43 |
+
0.999
|
| 44 |
+
],
|
| 45 |
+
"eps": 1e-08,
|
| 46 |
+
"weight_decay": 0.01,
|
| 47 |
+
"amsgrad": false,
|
| 48 |
+
"maximize": false,
|
| 49 |
+
"foreach": null,
|
| 50 |
+
"capturable": false,
|
| 51 |
+
"differentiable": false,
|
| 52 |
+
"fused": null,
|
| 53 |
+
"decoupled_weight_decay": true
|
| 54 |
+
}
|
| 55 |
+
],
|
| 56 |
+
"epoch_draws_from_public_sampler": 17532,
|
| 57 |
+
"archived_notebook_effective_batch": 10,
|
| 58 |
+
"expected_steps_four_epochs_batch10": 7016,
|
| 59 |
+
"inferred_checkpoint_compatible_effective_batch": 16,
|
| 60 |
+
"expected_steps_four_epochs_batch16": 4384,
|
| 61 |
+
"inference_is_not_recorded_original_argv": true,
|
| 62 |
+
"optimizer_updates_performed": 0
|
| 63 |
+
}
|