Registers a trainable or nontrainable parameter as a meta buffer. This can be later retrieved by meta_state_dict
(self, name: str, param: nn.Parameter)
| 106 | self._meta_params = set() |
| 107 | |
| 108 | def register_meta_buffer(self, name: str, param: nn.Parameter): |
| 109 | """ |
| 110 | Registers a trainable or nontrainable parameter as a meta buffer. This |
| 111 | can be later retrieved by meta_state_dict |
| 112 | """ |
| 113 | self.register_buffer(name, param) |
| 114 | self._meta_state_dict.add(name) |
| 115 | |
| 116 | def register_meta_parameter(self, name: str, parameter: nn.Parameter): |
| 117 | """ |
no outgoing calls
no test coverage detected