r""" Args: Save the current parameters for restoring later. parameters: Iterable of `torch.nn.Parameter`; the parameters to be temporarily stored.
(self, parameters: Iterable[torch.nn.Parameter])
| 527 | } |
| 528 | |
| 529 | def store(self, parameters: Iterable[torch.nn.Parameter]) -> None: |
| 530 | r""" |
| 531 | Args: |
| 532 | Save the current parameters for restoring later. |
| 533 | parameters: Iterable of `torch.nn.Parameter`; the parameters to be |
| 534 | temporarily stored. |
| 535 | """ |
| 536 | self.temp_stored_params = [param.detach().cpu().clone() for param in parameters] |
| 537 | |
| 538 | def restore(self, parameters: Iterable[torch.nn.Parameter]) -> None: |
| 539 | r""" |