(self, fabric: Fabric, *args: Any, **kwargs: Any)
| 343 | |
| 344 | class SpeedMonitorFabric(SpeedMonitorBase): |
| 345 | def __init__(self, fabric: Fabric, *args: Any, **kwargs: Any) -> None: |
| 346 | # TODO: this will not work properly if a precision plugin is passed to Fabric |
| 347 | flops_available = get_flops_available( |
| 348 | fabric.device, fabric._connector._precision_input |
| 349 | ) |
| 350 | super().__init__(flops_available, fabric.log_dict, *args, **kwargs) |
| 351 | |
| 352 | @fabric_rank_zero_only |
| 353 | def on_train_batch_end(self, *args: Any, **kwargs: Any): |
nothing calls this directly
no test coverage detected