(self, destination=None, prefix="", keep_vars=False, exclude_frozen_parameters=False)
| 3816 | return tensor_list |
| 3817 | |
| 3818 | def module_state_dict(self, destination=None, prefix="", keep_vars=False, exclude_frozen_parameters=False): |
| 3819 | sd = self.module.state_dict(destination=destination, prefix=prefix, keep_vars=keep_vars) |
| 3820 | |
| 3821 | # Remove frozen parameter weights from state_dict if specified |
| 3822 | if exclude_frozen_parameters: |
| 3823 | for n, p in self.module.named_parameters(): |
| 3824 | if not p.requires_grad and n in sd: |
| 3825 | del sd[n] |
| 3826 | |
| 3827 | if self.random_ltd_enabled(): |
| 3828 | sd = remove_random_ltd_state_dict(sd) |
| 3829 | return sd |
| 3830 | |
| 3831 | @staticmethod |
| 3832 | def _make_autoep_folding_metadata(folding_spec, |
no test coverage detected