def _is_quantiable_module(module):
    if isinstance(module, torch.quantization.FakeQuantize):
        return False
    return ((len(list(module.children())) > 0) or isinstance(module, Conv2d) or isinstance(module, Linear))
def _is_quantiable_module(module):
    if isinstance(module, torch.quantization.FakeQuantize):
        return False
    return ((len(list(module.children())) > 0) or isinstance(module, Conv2d) or isinstance(module, Linear))