Homework 6, with weight decay