Python on_step_training Examples

Example #1

    def test_on_step_training(self):
        def example(state):
            return state

        state = 'test'
        self.assertTrue(
            callbacks.on_step_training(example).on_step_training(state) ==
            state)

Example #2

File: test_decorators.py Project: little1tow/torchbearer

 def test_on_step_training(self):
     def example(state):
         return state
     state = 'test'
     self.assertTrue(callbacks.on_step_training(example).on_step_training(state) == state)