def get_printer(verbose, validation_label_letter):
if verbose >= 2:
printer = Tqdm(validation_label_letter=validation_label_letter)
elif verbose >= 1:
printer = Tqdm(validation_label_letter=validation_label_letter,
on_epoch=True)
else:
printer = Callback()
return printer
def test_state_dict(self):
callback = Callback()
self.assertEqual(callback.state_dict(), {})
self.assertEqual(callback.load_state_dict({}), callback)
def test_empty_methods(self):
callback = Callback()
self.assertIsNone(callback.on_start({}))
self.assertIsNone(callback.on_start_epoch({}))
self.assertIsNone(callback.on_start_training({}))
self.assertIsNone(callback.on_sample({}))
self.assertIsNone(callback.on_forward({}))
self.assertIsNone(callback.on_criterion({}))
self.assertIsNone(callback.on_backward({}))
self.assertIsNone(callback.on_step_training({}))
self.assertIsNone(callback.on_end_training({}))
self.assertIsNone(callback.on_end_epoch({}))
self.assertIsNone(callback.on_end({}))
self.assertIsNone(callback.on_start_validation({}))
self.assertIsNone(callback.on_sample_validation({}))
self.assertIsNone(callback.on_forward_validation({}))
self.assertIsNone(callback.on_end_validation({}))
self.assertIsNone(callback.on_step_validation({}))
self.assertIsNone(callback.on_criterion_validation({}))
def test_empty_methods(self):
callback = Callback()
self.assertIsNone(callback.on_start({}))
self.assertIsNone(callback.on_start_epoch({}))
self.assertIsNone(callback.on_start_training({}))
self.assertIsNone(callback.on_sample({}))
self.assertIsNone(callback.on_forward({}))
self.assertIsNone(callback.on_criterion({}))
self.assertIsNone(callback.on_backward({}))
self.assertIsNone(callback.on_step_training({}))
self.assertIsNone(callback.on_end_training({}))
self.assertIsNone(callback.on_end_epoch({}))
self.assertIsNone(callback.on_end({}))
self.assertIsNone(callback.on_start_validation({}))
self.assertIsNone(callback.on_sample_validation({}))
self.assertIsNone(callback.on_forward_validation({}))
self.assertIsNone(callback.on_end_validation({}))
self.assertIsNone(callback.on_step_validation({}))
self.assertIsNone(callback.on_criterion_validation({}))
