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