qualia_plugin_snn.learningmodel.pytorch.layers.spikingjelly.QuantizedGlobalSumPool1d module

Contain implementation of QuantizedGlobalSumPool1d layer with support for SpikingJelly step_mode.

class qualia_plugin_snn.learningmodel.pytorch.layers.spikingjelly.QuantizedGlobalSumPool1d.QuantizedGlobalSumPool1d[source]

Bases: QuantizedGlobalSumPool1d, StepModule

GlobalSumPool1d SpikingJelly’s step_mode support to Qualia’s QuantizedGlobalSumPool1d layer.

There is no need to override :meth:foward since it works the same for single-step or multi-step mode.

__init__(quant_params: QuantizationConfigDict, step_mode: str = 's') → None[source]

Construct QuantizedGlobalSumPool1d.

Parameters:

  • step_mode (str) – SpikingJelly’s step_mode, either 's' or 'm', see spikingjelly.activation_based.layer.Linear

  • quant_params (QuantizationConfigDict)

Return type:

None

extra_repr() → str[source]

GlobalSumPool1d step_mode to the __repr__ method.

Returns:

String representation of qualia_core.learningmodel.pytorch.layers.QuantizedGlobalSumPool1d with step_mode.

Return type:

str