【训练营】小模型训练支持 - #225

Open

ShenYouSOTA wants to merge 5 commits into

InfiniTensor:masterfrom

ShenYouSOTA:work/small-model-cnn-impl

Commits on Sep 18, 2026

feat: small-model CNN training support (Conv2d v1, ReLU, DataLoader, MNIST demo, DDP)
ShenYouSOTA
committed
a4fe470

Commits on Sep 19, 2026

fix: move ZeroGrad before DDP Forward in MNIST loop
opencode
committed
ed8d51a
docs: record P13b dual-GPU equivalence PASS and ZeroGrad fix
opencode
committed
b0307ec
fix: route MNIST loops through Module operator() and guard eval with NoGradGuard
opencode
committed
8770716
docs: restructure P13b result into tables with final dual-GPU confirmation
opencode
committed
b010284