Return the device of the model.
(self)
| 157 | |
| 158 | @property |
| 159 | def device(self) -> torch.device: |
| 160 | """ |
| 161 | Return the device of the model. |
| 162 | """ |
| 163 | return next(self.parameters()).device |
| 164 | |
| 165 | def convert_to(self, dtype: torch.dtype) -> None: |
| 166 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected