Unable to fuse convolution, batchnorm and relu.
Each entry in the module list looks something like
  (0): ConvBNReLU(
    (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
    (1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (2): ReLU6(inplace=True)
  )

At the time of fusion using below code i get the error.
 def fuse_model(self):
     for m in self.modules():
        if type(m) == ConvBNReLU:
           fuse_modules(m, ['0', '1', '2'], inplace=True)

NotImplementedError: Cannot fuse modules: (<class 'torch.nn.modules.conv.Conv2d'>, <class 'torch.nn.modules.batchnorm.BatchNorm2d'>, <class 'torch.nn.modules.activation.ReLU6'>)

Please let me know the way around.