MileTone_2 / environment /KNOWN_LIMITATIONS.md
SciCode's picture
Complete Milestone 2 pretrained models and training artifacts
4d5ab1c verified
|
Raw History Blame Contribute Delete
2.4 kB
# Environment Record and Known Reproducibility Limitations
## Run-Specific Evidence Available
The Qwen2.5-Coder-7B run has a contemporaneous package freeze, command, environment and
GPU snapshot under `evidence/training_runs/qwen25_coder_7b_cpt/`. Key versions were:
| Component | Version |
|---|---|
| ms-swift | 4.0.2 |
| PyTorch | 2.5.1+cu121 |
| Transformers | 5.3.0 |
| Accelerate | 1.13.0 |
| DeepSpeed | 0.18.8 |
| FlashAttention | 2.8.3, CUDA 12 / PyTorch 2.5 / CPython 3.11 wheel |
| GPUs | four NVIDIA H100 80 GB devices |
The run completed 18,165/18,165 steps and three epochs with recorded train loss 0.5602 and
token accuracy 0.8744. These are training telemetry, not downstream benchmark scores.
The GPT-2 and CodeGPT runs record the exact Docker image
`nvidia/cuda:12.2.2-devel-ubuntu22.04`, build command, effective trainer arguments, and the
clean llm.c Git commit. They do not include a Python package freeze because the trainer was
built and executed inside that CUDA development image.
## Known Gaps
The following May 2026 runs do not have their own complete package or system snapshots:
- `models/pretrained/stage1_cpt`
- `models/sft/sft_f3_refined_instruct`
Their checkpoint `args.json` files record effective ms-swift parameters, and their exact
datasets and historical launch scripts are included. A nearby April 2026 ms-swift snapshot
used the same key versions as the 7B run, but treating that as the exact May environment is
an inference. Bit-for-bit numerical retraining is not claimed.
The shared 20 GB raw dataset used by CodeGPT, GPT-2 and Qwen2.5-Coder-7B is not duplicated
in this package. Its historical source path, size and record count are recorded in the main
README. Reproducing those runs requires authorised access to that exact source data.
Raw llm.c model checkpoints and optimiser states are excluded after conversion to standard
Hugging Face checkpoints. Qwen optimiser, scheduler and RNG states are also excluded. The
delivered weights support inference and further training, but not exact mid-run resumption.
## Delivery Validation Environment
Validation uses the locally installed Python, PyTorch, Transformers and Safetensors
versions recorded in `evidence/smoke_test_results.json`. The smoke test checks that each
checkpoint is structurally loadable and produces finite, non-degenerate logits; it is not
a scientific performance evaluation.