Một bước optimizer đầy đủ

Mục tiêu kiến thức

Nhìn đủ ba bước zero_grad → backward → optimizer.step trong một update.

Mô tả bài toán

Đây là bài số 388 trong tầng PyTorch deep learning thực hành. Bài dùng thư viện loss.backward + optimizer.zero_grad + optimizer.step trên dữ liệu nhỏ, cố định và chạy CPU; mục tiêu là hiểu workflow chứ không bắt học viên tự viết lại thư viện.

Yêu cầu

Tạo Linear layer, tính MSE trước update, backward, lấy gradient norm, step một lần và tính loss sau update.

def single_training_step(X, y, learning_rate):

Ràng buộc

Seed 42; phải zero_grad trước backward và step đúng một lần.