File size: 159 Bytes
8cf6d22
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
import torch
import torch.nn


class DummyModel(torch.nn.Module):
    def __init__(self):
        super().__init__()

    def foward(self, x):
        return 1