# Codeblock 7b def forward(self, x): residual = x print(f'original\t\t: {x.size()}') x = self.conv0(x) print(f'after conv0\t\t: {x.size()}') x = self.conv1(x) print(f'after conv1\t\t: {x.size()}') x = self.semodule(x) print(f'after semodule\t\t: {x.size()}') x = self.conv2(x) print(f'after conv2\t\t: {x.size()}') if self.add: x += residual print(f'after summation\t\t: {x.size()}') return x