Toggle @pytorch.jit.script decorator on / off

1

Hello,

is there a way to toggle the toggle @pytorch.jit.script decorator on / off, like:

use_jit = True / False
@torch.jit.script(enable=use_jit)
def function(x)
    pass

Thanks!

1 Like