[jacinto-ai/pytorch-jacinto-ai-devkit.git] / modules / pytorch_jacinto_ai / xnn / quantize / quant_graph_module.py
diff --git a/modules/pytorch_jacinto_ai/xnn/quantize/quant_graph_module.py b/modules/pytorch_jacinto_ai/xnn/quantize/quant_graph_module.py
index ba907754d42dc76c0e2ce80ce0b60f924d262f19..d4f5f7d9d89d651e5197599cb77f7933b028d682 100644 (file)
activation_q = layers.PAct2(signed=False)
elif isinstance(module, torch.nn.Hardtanh):
activation_q = layers.PAct2(clip_range=(module.min_val, module.max_val))
+ elif isinstance(module, layers.NoAct):
+ activation_q = layers.PAct2(signed=None)
else:
activation_q = layers.PAct2(signed=None)
#