(self,x)
| 9 | self.conv2 = nn.Conv2d(2,3,3) |
| 10 | |
| 11 | def forward(self,x): |
| 12 | b,c,h,w = x.shape |
| 13 | if c ==1: |
| 14 | x = self.conv1(x) |
| 15 | else: |
| 16 | x = self.conv2(x) |
| 17 | return x |
| 18 | |
| 19 | model = MyModule() |
| 20 |
nothing calls this directly
no outgoing calls
no test coverage detected