In optimizer.zero_grad(), set p.grad = None?

Double post from here.