I want to add after 5th layer that is after
   nn.Conv2d(16, 16, kernel_size=3, stride=1, padding=1, bias=True),   
   nn.BatchNorm2d(16),
   nn.Dropout2d(p=0.5, inplace=False),
   nn.ReLU6(inplace=True)