Gradient Noise Scale and the Critical Batch Size

medium
Unit 19 · Memory0/2 to milestone

How big should the batch be? McCandlish et al. (2018) showed that the useful batch size is set by the gradient noise scale

Pro problem

Unlock this and every premium problem across all tracks — plus hidden-test grading, advanced AI engineering modules and new content every week.

Start 7-day free trial
solution.py · Python 3.13
Run your code to see results. ⌘↵ to run · ⌘⇧↵ to submit
Gradient Noise Scale and the Critical Batch Size · TechGradify