During static quantization of my model, I encounter the following error -
RuntimeError: Didn’t find kernel to dispatch to for operator ‘aten::_cat’. Tried to look up kernel for dispatch key ‘QuantizedCPUTensorId’. Registered dispatch keys are: [CPUTensorId, VariableTensorId]
I have fused and quantized the model, as well as the input image. But it throws an error on concat raised by - y = torch.cat([sources[0], sources[1]], dim=1)
Any suggestions would be appreciated.   
Full code here -

  
      github.com
  
  
    raghavgurbaxani/experiments/blob/master/try_static_quant.py 16
###  CRAFT model here - https://github.com/clovaai/CRAFT-pytorch/blob/master/craft.py
import torch
import os
import time
from craft import CRAFT
import cv2
import numpy as np
import craft_utils
import imgproc
import file_utils
from torch.autograd import Variable
canvas_size=1280
mag_ratio=1.5
trained_model='craft_mlt_25k.pth'
def uninplace(model):
    if hasattr(model, 'inplace'):
        model.inplace = False
    if not model.children():
        return
    for child in model.children():


  This file has been truncated. show original