But I get another error now:
RuntimeError: OrderedDict mutated during iteration

I am looping over all the parameters of my (potentially arbitrary) model. So, I guess python gets unhappy that I’ve mutated it…