Print architecture summary.
(self)
| 127 | pass # REPLACE |
| 128 | |
| 129 | def __repr__(self) -> str: |
| 130 | """Print architecture summary.""" |
| 131 | total_params = sum(p.numel() for p in self.parameters()) |
| 132 | trainable_params = sum(p.numel() for p in self.parameters() if p.requires_grad) |
| 133 | return ( |
| 134 | f"{self.__class__.__name__}(\n" |
| 135 | f" config={self.config},\n" |
| 136 | f" total_params={total_params:,},\n" |
| 137 | f" trainable_params={trainable_params:,}\n" |
| 138 | f")" |
| 139 | ) |
nothing calls this directly
no outgoing calls
no test coverage detected