Log scalar data to TensorBoard. Args: name: Tag name used to group scalars data: Scalar data to log (float/int/Tensor) step: Step value to record
(self, name: str, data: Any, step: int)
| 96 | self.log(key, value, step) |
| 97 | |
| 98 | def log(self, name: str, data: Any, step: int) -> None: |
| 99 | """Log scalar data to TensorBoard. |
| 100 | |
| 101 | Args: |
| 102 | name: Tag name used to group scalars |
| 103 | data: Scalar data to log (float/int/Tensor) |
| 104 | step: Step value to record |
| 105 | """ |
| 106 | if not self._writer: |
| 107 | return |
| 108 | |
| 109 | self._writer.add_scalar(name, data, global_step=step, new_style=True) |
| 110 | |
| 111 | def log_visuals( |
| 112 | self, |
no outgoing calls
no test coverage detected