# Codeblock 4b
    def forward(self, x):
        print(f'original\t: {x.size()}')
        
        print()
        for i in range(len(self.stage0)):
            x = self.stage0[i](x)
            print(f'after stage0 #{i}\t: {x.size()}')
        
        print()
        for i in range(len(self.stage1)):
            x = self.stage1[i](x)
            print(f'after stage1 #{i}\t: {x.size()}')
        
        print()
        for i in range(len(self.stage2)):
            x = self.stage2[i](x)
            print(f'after stage2 #{i}\t: {x.size()}')
        
        print()
        for i in range(len(self.stage3)):
            x = self.stage3[i](x)
            print(f'after stage3 #{i}\t: {x.size()}')
        
        print()
        for i in range(len(self.stage4)):
            x = self.stage4[i](x)
            print(f'after stage4 #{i}\t: {x.size()}')
            
        x_stage4 = x.clone()        #(1)
        
        print()
        for i in range(len(self.stage5)):
            x = self.stage5[i](x)
            print(f'after stage5 #{i}\t: {x.size()}')
        
        x_stage5 = x.clone()        #(2)

        return x_stage4, x_stage5   #(3)