Copy the master parameter data back into the model parameters.
(model_params, master_params)
| 48 | |
| 49 | |
| 50 | def master_params_to_model_params(model_params, master_params): |
| 51 | """ |
| 52 | Copy the master parameter data back into the model parameters. |
| 53 | """ |
| 54 | # Without copying to a list, if a generator is passed, this will |
| 55 | # silently not copy any parameters. |
| 56 | model_params = list(model_params) |
| 57 | |
| 58 | for param, master_param in zip( |
| 59 | model_params, unflatten_master_params(model_params, master_params) |
| 60 | ): |
| 61 | param.detach().copy_(master_param) |
| 62 | |
| 63 | |
| 64 | def unflatten_master_params(model_params, master_params): |
no test coverage detected