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({}))