pshashk:

tq.DeQuantStub()


We actually don’t have relu6 in the mapping right now: https://github.com/pytorch/pytorch/blob/master/torch/quantization/default_mappings.py#L14 111. adding an entry here should fix the problem. Thanks for reporting.