Hello. Your solution is correct.
But I met the problem because the model has to much embeddings like  “module.mixed_stem2.branch0.0.bn.weight”
So I always get wrong when I use your method because the syntax of “batch0.0”
What should I do to iter the weight and bias of this model so I can set different learning rate for them?