nni GELU is not registered, please register setting with QuantizationSetting.register().

[mikhilg10]

getting error
AssertionError: GELU is not registered, please register setting with QuantizationSetting.register()
tried with
from nni.compression.quantization import QuantizationSetting
but still no module availabe from compression it gives.

[mikhilg10]

i have tried to add the gelu activation setting still getting error
`import nni
from nni.compression.pytorch.utils import QuantizationSetting

activation_setting = {
'input': {
'quant_dtype': None,
'quant_scheme': None,
'granularity': 'default',
'apply_method': 'clamp_round',
},
'output': {
'quant_dtype': None,
'quant_scheme': None,
'granularity': 'default',
'apply_method': 'clamp_round',
}
}

QuantizationSetting.register(torch.nn.GELU(), activation_setting)`