Skip to content

README.md

Latest commit

 

History

History
32 lines (24 loc) · 689 Bytes

README.md

File metadata and controls

32 lines (24 loc) · 689 Bytes
Raw

mobile-block Build Status PyPI version

MobileNet-style block for PyTorch

Installation

pip install mobile-block

Usage

Require Python 3.6+, torch 0.4+.

from mobile_block import MobileBlock

# normal usage
block = MobileBlock(
  input_size=112,
  in_channels=16,
  out_channels=32,
  stride=1,
  expansion=6,
  kernel=3,
  groups=1
)

# factory usage
mock_block_id = 'w112_i16_o32_s1_e6_k3_g1'
block = MobileBlock.factory(mock_block_id)