(self, inputBatch)
| 35 | |
| 36 | |
| 37 | def forward(self, inputBatch): |
| 38 | batch = F.relu(self.bn1a(self.conv1a(inputBatch))) |
| 39 | batch = self.conv2a(batch) |
| 40 | if self.stride == 1: |
| 41 | residualBatch = inputBatch |
| 42 | else: |
| 43 | residualBatch = self.downsample(inputBatch) |
| 44 | batch = batch + residualBatch |
| 45 | intermediateBatch = batch |
| 46 | batch = F.relu(self.outbna(batch)) |
| 47 | |
| 48 | batch = F.relu(self.bn1b(self.conv1b(batch))) |
| 49 | batch = self.conv2b(batch) |
| 50 | residualBatch = intermediateBatch |
| 51 | batch = batch + residualBatch |
| 52 | outputBatch = F.relu(self.outbnb(batch)) |
| 53 | return outputBatch |
| 54 | |
| 55 | |
| 56 |
nothing calls this directly
no outgoing calls
no test coverage detected