def test_trainer_can_log_learning_rates_tensorboard(self):
data_loader = DataLoader(self.instances, batch_size=4, collate_fn=allennlp_collate)
trainer = GradientDescentTrainer(
self.model,
self.optimizer,
data_loader,
num_epochs=2,
serialization_dir=self.TEST_DIR,
tensorboard_writer=TensorboardWriter(
serialization_dir=self.TEST_DIR, should_log_learning_rate=True, summary_interval=2,
),
)
trainer.train()
def test_trainer_can_log_histograms(self):
# enable activation logging
for module in self.model.modules():
module.should_log_activations = True
trainer = GradientDescentTrainer(
self.model,
self.optimizer,
self.data_loader,
num_epochs=3,
serialization_dir=self.TEST_DIR,
tensorboard_writer=TensorboardWriter(
serialization_dir=self.TEST_DIR, histogram_interval=2),
)
trainer.train()
