# Codeblock 7c def forward(self, x): print(f'original\t\t\t: {x.size()}') x_stage4, x_stage5 = self.darknet(x) #(1) print(f'\nx_stage4\t\t\t: {x_stage4.size()}') print(f'x_stage5\t\t\t: {x_stage5.size()}') print() x = x_stage5 for i in range(len(self.stage6)): x = self.stage6[i](x) #(2) print(f'x_stage5 after stage6 #{i}\t: {x.size()}') x_stage4 = self.passthrough(x_stage4) #(3) print(f'\nx_stage4 after passthrough\t: {x_stage4.size()}') x_stage4 = self.reorder(x_stage4) #(4) print(f'x_stage4 after reorder\t\t: {x_stage4.size()}') x = torch.cat([x_stage4, x], dim=1) #(5) print(f'\nx after concatenate\t\t: {x.size()}') for i in range(len(self.stage7)): #(6) x = self.stage7[i](x) print(f'x after stage7 #{i}\t: {x.size()}') return x