def test_auto_shoud_be_max(self, _): state = {torchbearer.METRICS: {'acc_loss': 0.1}} file_path = 'test_file_{acc_loss:.2f}' check = Best(file_path, monitor='acc_loss') check.on_start(state) check.on_end_epoch(state) self.assertTrue(check.mode == 'max')
def test_auto_shoud_be_max(self, _): state = {torchbearer.METRICS: {'acc_loss': 0.1}} file_path = 'test_file_{acc_loss:.2f}' check = Best(file_path, monitor='acc_loss') check.on_start(state) check.on_end_epoch(state) self.assertTrue(check.mode == 'max')
def test_min_delta_save(self, mock_save): state = {torchbearer.METRICS: {'val_loss': 0.1}} file_path = 'test_file_{val_loss:.2f}' check = Best(file_path, mode='min', min_delta=0.1) check.on_start(state) check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 1) state = {torchbearer.METRICS: {'val_loss': -0.001}} check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 2)
def test_max_with_decreasing(self, mock_save): state = {torchbearer.METRICS: {'val_loss': 0.1}} file_path = 'test_file_{val_loss:.2f}' check = Best(file_path, mode='max') check.on_start(state) check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 1) state = {torchbearer.METRICS: {'val_loss': 0.001}} check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 1)
def test_min_delta_save(self, mock_save): state = {torchbearer.METRICS: {'val_loss': 0.1}} file_path = 'test_file_{val_loss:.2f}' check = Best(file_path, mode='min', min_delta=0.1) check.on_start(state) check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 1) state = {torchbearer.METRICS: {'val_loss': -0.001}} check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 2)
def test_max_with_decreasing(self, mock_save): state = {torchbearer.METRICS: {'val_loss': 0.1}} file_path = 'test_file_{val_loss:.2f}' check = Best(file_path, mode='max') check.on_start(state) check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 1) state = {torchbearer.METRICS: {'val_loss': 0.001}} check.on_end_epoch(state) self.assertTrue(mock_save.call_count == 1)