Integrations
Tensorboard
Using VESSL’s Python SDK, you can view and interact with metrics and media logged to TensorBoard directly on VESSL. We currently support scalars, images, and audio.
VESSL supports TensorBoard with TensorFlow, PyTorch (TensorBoard > 1.14), and TensorBoardX.
You can integrate TensorBoard by simply adding vessl.init(tensorboard=True)
to your code.
Note that this should be called before creating the file writer. This is because VESSL auto-detects the TensorBoard logdir
upon writer creation but cannot do so if the writer has already been created.
TensorFlow
import tensorflow as tf
import vessl
vessl.init(tensorboard=True) # Must be called before tf.summary.create_file_writer
w = tf.summary.create_file_writer("./logdir")
...
PyTorch
from torch.utils.tensorboard import SummaryWriter
import vessl
vessl.init(tensorboard=True) # Must be called before SummaryWriter
writer = SummaryWriter("newdir")
...
TensorBoardX
from tensorboardX import SummaryWriter
import vessl
vessl.init(tensorboard=True) # Must be called before SummaryWriter
writer = SummaryWriter("newdir")
...