Ok, because I got this error here
----> return sum(p for p in model.parameters() if p.requires_grad)

RuntimeError: The size of tensor a (7) must match the size of tensor b (64) at non-singleton dimension 3