(self, model)
| 48 | assert not key in self.m_name2s_name |
| 49 | |
| 50 | def copy_to(self, model): |
| 51 | m_param = dict(model.named_parameters()) |
| 52 | shadow_params = dict(self.named_buffers()) |
| 53 | for key in m_param: |
| 54 | if m_param[key].requires_grad: |
| 55 | m_param[key].data.copy_(shadow_params[self.m_name2s_name[key]].data) |
| 56 | else: |
| 57 | assert not key in self.m_name2s_name |
| 58 | |
| 59 | def store(self, parameters): |
| 60 | """ |