Unflatten the master parameters to look like model_params.
(model_params, master_params)
| 62 | |
| 63 | |
| 64 | def unflatten_master_params(model_params, master_params): |
| 65 | """ |
| 66 | Unflatten the master parameters to look like model_params. |
| 67 | """ |
| 68 | return _unflatten_dense_tensors(master_params[0].detach(), tuple(tensor for tensor in model_params)) |
| 69 | |
| 70 | |
| 71 | def zero_grad(model_params): |
no outgoing calls
no test coverage detected