# Codeblock 7b def forward(self, x): print(f'original\t\t: {x.size()}') x = self.relu(self.resnext_bn1(self.resnext_conv1(x))) print(f'after resnext_conv1\t: {x.size()}') x = self.resnext_maxpool1(x) print(f'after resnext_maxpool1\t: {x.size()}') for i, block in enumerate(self.resnext_conv2): x = block(x) print(f'after resnext_conv2 #{i}\t: {x.size()}') for i, block in enumerate(self.resnext_conv3): x = block(x) print(f'after resnext_conv3 #{i}\t: {x.size()}') for i, block in enumerate(self.resnext_conv4): x = block(x) print(f'after resnext_conv4 #{i}\t: {x.size()}') for i, block in enumerate(self.resnext_conv5): x = block(x) print(f'after resnext_conv5 #{i}\t: {x.size()}') x = self.avgpool(x) print(f'after avgpool\t\t: {x.size()}') x = torch.flatten(x, start_dim=1) print(f'after flatten\t\t: {x.size()}') x = self.fc(x) print(f'after fc\t\t: {x.size()}') return x