Welcome to bitorch’s documentation!

BITorch is a library currently under development to simplify building quantized and binary neural networks with PyTorch. This is an early preview version of the library. If you wish to use it and encounter any problems, please create an issue. Our current roadmap contains:

• Extending the model zoo with pre-trained models of state-of-the-art approaches

• Adding examples for advanced training methods with multiple stages, knowledge distillation, etc.

All changes are tracked in the changelog.

Contents:

• Installation
  • Pip
  • Local and Development Install Options
  • Dali Preprocessing
  • Code formatting and typing
• Build Docs
• Introduction
  • Binary Neural Networks
  • Imports
• Full Documentation
  • bitorch
    • bitorch.add_config_args
      • add_config_args()
    • bitorch.apply_args_to_configuration
      • apply_args_to_configuration()
    • bitorch.config_from_name
      • config_from_name()
    • bitorch.config_names
      • config_names()
    • bitorch.register_custom_config
      • register_custom_config()
    • bitorch.config
      • bitorch.config.Config
    • bitorch.layers
      • bitorch.layers.convert
      • bitorch.layers.bembedding
      • bitorch.layers.config
      • bitorch.layers.debug_layers
      • bitorch.layers.extensions
      • bitorch.layers.pact
      • bitorch.layers.pad
      • bitorch.layers.qactivation
      • bitorch.layers.qconv1d
      • bitorch.layers.qconv2d
      • bitorch.layers.qconv3d
      • bitorch.layers.qconv_mixin
      • bitorch.layers.qembedding
      • bitorch.layers.qlinear
      • bitorch.layers.register
    • bitorch.models
      • bitorch.models.model_from_name
      • bitorch.models.model_names
      • bitorch.models.register_custom_model
      • bitorch.models.base
      • bitorch.models.common_layers
      • bitorch.models.densenet
      • bitorch.models.dlrm
      • bitorch.models.lenet
      • bitorch.models.meliusnet
      • bitorch.models.model_hub
      • bitorch.models.quicknet
      • bitorch.models.resnet
      • bitorch.models.resnet_e
    • bitorch.quantizations
      • bitorch.quantizations.quantization_from_name
      • bitorch.quantizations.quantization_names
      • bitorch.quantizations.register_custom_quantization
      • bitorch.quantizations.approx_sign
      • bitorch.quantizations.base
      • bitorch.quantizations.config
      • bitorch.quantizations.dorefa
      • bitorch.quantizations.identity
      • bitorch.quantizations.progressive_sign
      • bitorch.quantizations.quantization_scheduler
      • bitorch.quantizations.sign
      • bitorch.quantizations.ste_heaviside
      • bitorch.quantizations.swish_sign
    • bitorch.runtime_mode
      • bitorch.runtime_mode.RuntimeMode
      • bitorch.runtime_mode.change_mode
      • bitorch.runtime_mode.pause_wrapping
    • bitorch.util
      • bitorch.util.build_lookup_dictionary

Indices and tables

• Index

• Module Index

• Search Page