API Reference

Available Modules

• nn
  • MemSaveConv2d
  • MemSaveLinear
  • MemSaveReLU
  • MemSaveMaxPool2d
  • MemSaveBatchNorm2d
  • MemSaveLayerNorm
• util ⚠️
  • ResultsCollector
  • estimate.py
  • Models

memsave_torch package

The memsave_torch package consists of one main module: nn