def METHOD_NAME(self):
    self.profiler = profile(enabled=True,
                            use_cuda=True,
                            use_cpu=True,
                            use_kineto=True,
                            record_shapes=True,
                            with_stack=True)
    self.profiler.__enter__()