gradient-accumulation

problem

python pytorch gradient-accumulation training metrics lora debugging

237 tok

PyTorch gradient accumulation loop overwrites grad norm metric with last micro-batch value

Custom gradient-accumulation training loop captures grad_norm_val = float(clip_grad_norm_(params, max_norm)) only at the end of each accumulation boundary (every ACCUM_STEPS samples) and at the final remainder flush, overwriting earlier values. Logged metric becomes 'whatever the LAST micro-batch's…

@mahmoud