objax.jaxboard package¶
|
Reduces tensor batch into a single tensor. |
Writes entries to Summary protocol buffer. |
|
|
Writes entries to event files in the logdir to be consumed by Tensorboard. |
- class objax.jaxboard.Summary[source]¶
Writes entries to Summary protocol buffer.
- image(tag, image)[source]¶
Adds image to the summary. Float image in [-1, 1] in CHW format expected.
- Parameters
tag (str) –
image (ndarray) –
- class objax.jaxboard.SummaryWriter(logdir, queue_size=5, write_interval=5)[source]¶
Writes entries to event files in the logdir to be consumed by Tensorboard.
- Parameters
logdir (str) –
queue_size (int) –
write_interval (int) –
- __init__(logdir, queue_size=5, write_interval=5)[source]¶
Creates SummaryWriter instance.
- Parameters
logdir (str) – directory where event file will be written.
queue_size (int) – size of the queue for pending events and summaries before one of the ‘add’ calls forces a flush to disk.
write_interval (int) – how often, in seconds, to write the pending events and summaries to disk.