(self, path)
| 19 | |
| 20 | class AudioEncoder(nn.Module): |
| 21 | def __init__(self, path): |
| 22 | super().__init__() |
| 23 | self.model = torch.jit.load(path) |
| 24 | self.register_buffer('hidden', torch.zeros(2, 1, 256)) |
| 25 | |
| 26 | def forward(self, audio): |
| 27 | self.reset() |
nothing calls this directly
no outgoing calls
no test coverage detected