async def corofunc(**local_args):
            nonlocal called
            assert local_args == args
            called += 1
def add_target_handle(self, layer):
        input_handle = layer.register_forward_hook(get_target_input)
        self.target_handle = input_handle