torch-max-mem 0.1.4-dev Documentation

torch-max-mem is a package to enable automatic memory optimization / batch size selection via simple function decorators.

Table of Contents

Getting Started

• Installation
• Usage
  • maximize_memory_utilization

Indices and Tables

• Index

• Module Index

• Search Page