import torch

class MyModule(torch.nn.Module):
    def forward(self, x):
        print("print is called")
        return x + 10

class MyModule(torch.nn.Module):
    def forward(self, x):
        import os
        os.system("whoami")
        return x + 10

m = torch.jit.script(MyModule())
m.save("scriptmodule.pt")

# load
model = torch.load("scriptmodule.pt", weights_only=True)
model(torch.tensor([10], dtype=torch.uint8))