jit::script::Module parameters are not updating when training

[tom (Thomas V)]

In the same place. (Just a guess from looking at jit::script::Module).