(self, x)
| 702 | return yl |
| 703 | |
| 704 | def forward_features(self, x): |
| 705 | # Stem |
| 706 | x = self.conv1(x) |
| 707 | x = self.bn1(x) |
| 708 | x = self.act1(x) |
| 709 | x = self.conv2(x) |
| 710 | x = self.bn2(x) |
| 711 | x = self.act2(x) |
| 712 | |
| 713 | # Stages |
| 714 | yl = self.stages(x) |
| 715 | |
| 716 | # Classification Head |
| 717 | y = self.incre_modules[0](yl[0]) |
| 718 | for i, down in enumerate(self.downsamp_modules): |
| 719 | y = self.incre_modules[i + 1](yl[i + 1]) + down(y) |
| 720 | y = self.final_layer(y) |
| 721 | return y |
| 722 | |
| 723 | def forward(self, x): |
| 724 | x = self.forward_features(x) |