a894f84b099fb2ab1dee9ebce09361112f2539c2,docs/_static/examples/visdom_note.py,,,#,52

Before Change

torchbearer_model.fit_generator(traingen, epochs=5, validation_generator=valgen,
                                callbacks=[TensorBoard(visdom=True, write_graph=False, write_batch_metrics=True, batch_step_size=10, write_epoch_metrics=False)])

torchbearer_model.fit_generator(traingen, epochs=5, validation_generator=valgen,
                                callbacks=[TensorBoard(visdom=True, write_graph=False, write_batch_metrics=False, write_epoch_metrics=True)])

import torchbearer.callbacks.tensor_board as tensorboard

tensorboard.VisdomParams.ENV = "Test"

After Change

from torchbearer import Trial
from torchbearer.callbacks import TensorBoard

torchbearer_trial = Trial(model, optimizer, loss, metrics=["acc", "loss"], callbacks=[TensorBoard(visdom=True, write_graph=True, write_batch_metrics=True, batch_step_size=10, write_epoch_metrics=True)]).to("cuda")
torchbearer_trial.with_generators(train_generator=traingen, val_generator=valgen)
torchbearer_trial.run(epochs=5)

import torchbearer.callbacks.tensor_board as tensorboard

tensorboard.VisdomParams.ENV = "Test"

Instances