TensorBoard

Trainer callback that logs training, validation, and test diagnostics to TensorBoard and serves them, so a run can be monitored in the browser as it progresses. Logs are written under the run’s output directory.

Configuration

Option

Description

Type

Default

port

Port the TensorBoard server is served on.

int

required

trainer.register_callback(CallbackSpec(callback=TensorBoardCallback, kwargs={"port": 6006}))