bitorch.models.meliusnet.MeliusNetFlex¶

class bitorch.models.meliusnet.MeliusNetFlex(*args: Any, **kwargs: Any)[source]¶

MeliusNet-Flex model from “MeliusNet: Can Binary Neural Networks Achieve MobileNet-level Accuracy?” <https://arxiv.org/abs/2001.05936> paper.

Methods

`__init__`	Initializes internal Module state, shared by both nn.Module and ScriptModule.
`add_argparse_arguments`	allows additions to the argument parser if required, e.g.

Attributes

name

__init__(*args: Any, **kwargs: Any) → None[source]¶: Initializes internal Module state, shared by both nn.Module and ScriptModule.

static add_argparse_arguments(parser: ArgumentParser) → None[source]¶

allows additions to the argument parser if required, e.g. to add layer count, etc.

! please note that the inferred variable names of additional cli arguments are passed as keyword arguments to the constructor of this class !

bitorch