# Codeblock 7b def forward(self, x): print(f'original\t\t: {x.size()}') x = self.norm1(self.relu(self.conv1(x))) print(f'after conv1\t\t: {x.size()}') x = self.pool1(x) print(f'after pool1\t\t: {x.size()}') x = self.norm2(self.relu(self.conv2(x))) print(f'after conv2\t\t: {x.size()}') x = self.pool2(x) print(f'after pool2\t\t: {x.size()}') x = self.relu(self.conv3(x)) print(f'after conv3\t\t: {x.size()}') x = self.relu(self.conv4(x)) print(f'after conv4\t\t: {x.size()}') x = self.relu(self.conv5(x)) print(f'after conv5\t\t: {x.size()}\n') x = self.spp(x) print(f'after spp\t\t: {x.size()}\n') x = self.dropout6(self.relu(self.fc6(x))) print(f'after fc6\t\t: {x.size()}') x = self.dropout7(self.relu(self.fc7(x))) print(f'after fc7\t\t: {x.size()}') x = self.fc8(x) print(f'after fc8\t\t: {x.size()}') return x