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

Before Change

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

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

After Change

torchbearer_trial.with_generators(train_generator=traingen, val_generator=valgen)
torchbearer_trial.run(epochs=10)

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

Instances