Segment 54

In this sub-session, the user pivoted from architecture tuning to data-centric improvements. They set up SGLang on SM120 for high-throughput batch inference, generating 193K diverse prompts from Infinity-Instruct, WebInstruct, CodeFeedback, MetaMathQA, Hermes FC, and Agent Training datasets, producing 523M output tokens. These were tokenized and merged with the existing 902K dataset, yielding 1,095,082 samples totaling 2.411B tokens. Training was resumed from step 690 with the expanded dataset, but GPU 6 OOM'd during ramp-up. Attempts to restart with reduced memory parameters (token_budget 45056, max_batch_size 48) and a 6-target + 2-drafter configuration failed to recover the previous 20 Ktok/s throughput. The user rejected the degraded performance, and the assistant began reverting torch from cu130 back to cu128 to restore the original memory budget and stable 5t+3d configuration.

Set up SGLang on SM120 for batch inferenceGenerate 193K diverse prompts from multiple datasetsTokenize and merge new data with existing datasetResume DFlash training from step 690Diagnose GPU 6 OOM during training ramp-upReduce token_budget and max_batch_size to fit memorySwitch to 6-target + 2-drafter GPU topologyRevert torch from cu130 to cu128 to restore memory budget

The 200MB That Broke a Training Pipeline: Data Expansion, Dependency Cascades, and the Fragility of GPU Memory Budgets on 8× Blackwell GPUs 4116 words

Chunks