spikingjelly.activation_based.memopt package

用于自定义梯度检查点结构和应用论文预设的训练显存优化工具。

Training-memory tools for custom checkpoint structures and the paper preset.

Public API

checkpoint	Checkpoint one callable and optionally compress its first tensor input.
checkpoint_module	Wrap one module transparently, with optional temporal chunks.
optimize_memory	Apply the high-level progressive checkpoint preset.

Spike Compressors

将浮点数表示的脉冲张量转换为更紧凑的表示形式的压缩器。

---

Compressors that convert spike tensors represented in floating-point numbers into more compact representations.

SpikeCompressor	Abstract base class for stateless compressors.
BooleanSpikeCompressor	Convert spike tensors to/from boolean tensors.
Uint8SpikeCompressor	Convert spike tensors to/from uint8 tensors.
BitSpikeCompressor	Converts spike tensors to/from bit representations.
SparseSpikeCompressor	Convert spike tensors to/from sparse representations.