8cf6d22 ce2dcfc 14e27af db65a6c
1
2
3
4
5
6
7
8
9
10
11
12
13
import torch import torch.nn class DummyModel(torch.nn.Module): def __init__(self): super().__init__() def forward(self, x): return torch.softmax(torch.rand(5), 0) def __call__(self, x): return self.forward(x)
