(module: nn.Module, persistent: bool = True)
| 2 | |
| 3 | |
| 4 | def convert_to_buffer(module: nn.Module, persistent: bool = True): |
| 5 | # Recurse over child modules. |
| 6 | for name, child in list(module.named_children()): |
| 7 | convert_to_buffer(child, persistent) |
| 8 | |
| 9 | # Also re-save buffers to change persistence. |
| 10 | for name, parameter_or_buffer in ( |
| 11 | *module.named_parameters(recurse=False), |
| 12 | *module.named_buffers(recurse=False), |
| 13 | ): |
| 14 | value = parameter_or_buffer.detach().clone() |
| 15 | delattr(module, name) |
| 16 | module.register_buffer(name, value, persistent=persistent) |
no outgoing calls
no test coverage detected