(model)
| 213 | |
| 214 | |
| 215 | def zero_init_global_gradient(model): |
| 216 | grads = dict() |
| 217 | for name, param in model.named_parameters(): |
| 218 | param.requires_grad_(True) |
| 219 | grads[name] = 0 |
| 220 | return grads |
| 221 | |
| 222 | |
| 223 | def data_mix(ori_data, pse_data, mix_ratio): |
nothing calls this directly
no outgoing calls
no test coverage detected