Crate bimm 

bimm - Burn Image Models

Notable Components

• cache - weight loading cache.
• compat - compat code, ported or planned for an upcoming release of burn.
• layers - reusable neural network modules.
  • layers::blocks - miscellaneous blocks.
    • layers::blocks::cna::CNA2d - Conv2d + Normalization + Activation block.
    • layers::blocks::conv_norm::ConvNorm2d - Conv2d + BatchNorm2d block.
  • layers::drop - dropout layers.
    • layers::drop::drop_block - 2d drop block / spatial dropout.
    • layers::drop::drop_path - drop path / stochastic depth.
  • layers::patching - patching layers.
    • layers::patching::patch_embed - 2d patch embedding layer.
• models - complete model families.
  • models::resnet - ResNet
  • models::swin - The SWIN Family.
    • models::swin::v2 - The SWIN-V2 Model.

Modules

cache

Cache Implementations

compat

Compat code to bridge burn_support updates.

layers

Common low-level modules for adding layers and operations in Burn.

models

Common high-level modules for building models in Burn.

utility

Utility Support Functions