Solved by jerryzh168 in post #2 
              
              
                I think you need the following: 
def fuse_model(self):
      torch.quantization.fuse_modules(self, modules_to_fuse=[["linear", "relu"]], inplace=True)