| 3 | from functions.SeparableConvLayer import SeparableConvLayer |
| 4 | |
| 5 | class SeparableConvModule(Module): |
| 6 | def __init__(self,filtersize): |
| 7 | super(SeparableConvModule, self).__init__() |
| 8 | self.f = SeparableConvLayer(filtersize) |
| 9 | |
| 10 | def forward(self, input1, input2, input3): |
| 11 | return self.f(input1, input2, input3) |
| 12 | |
| 13 | #we actually dont need to write the backward code for a module, since we have |
| 14 |
no outgoing calls