File size: 219 Bytes
8cf6d22
 
 
 
 
 
 
 
ce2dcfc
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 1

    def __call__(self, x):
        return self.forward(x)