(self, destination=None, prefix="", keep_vars=False, exclude_frozen_parameters=False)
| 3482 | return tensor_list |
| 3483 | |
| 3484 | def module_state_dict(self, destination=None, prefix="", keep_vars=False, exclude_frozen_parameters=False): |
| 3485 | sd = self.module.state_dict(destination=destination, prefix=prefix, keep_vars=keep_vars) |
| 3486 | |
| 3487 | # Remove frozen parameter weights from state_dict if specified |
| 3488 | if exclude_frozen_parameters: |
| 3489 | for n, p in self.module.named_parameters(): |
| 3490 | if not p.requires_grad and n in sd: |
| 3491 | del sd[n] |
| 3492 | |
| 3493 | if self.random_ltd_enabled(): |
| 3494 | sd = remove_random_ltd_state_dict(sd) |
| 3495 | return sd |
| 3496 | |
| 3497 | @staticmethod |
| 3498 | def load_moe_state_dict(checkpoint_path, |
no test coverage detected