Keep in mind
-
Shapes:
X @ W + bproduces predictions with the same shape as targetsY. -
Learning: Each SGD step samples one example and updates
Wandbby subtractinglearning_rate * gradient. The data stays fixed. -
Comparisons: Learning-rate comparisons hold the initial weights, biases, and sampled examples fixed. Each comparison starts training from those same initial parameters.
-
Progress:
loss_historyrecords full-data MSE before training and after each of ten updates. Lower is better, but individual SGD steps can increase it.