bitorch.models.densenet.DenseNetFlex

class bitorch.models.densenet.DenseNetFlex(*args: Any, **kwargs: Any)

Flexible BinaryDenseNet model from “BinaryDenseNet: Developing an Architecture for Binary Neural Networks” <https://openaccess.thecvf.com/content_ICCVW_2019/html/NeurArch/Bethge_BinaryDenseNet_Developing_an_Architecture_for_Binary_Neural_Networks_ICCVW_2019_paper.html> 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

Initializes internal Module state, shared by both nn.Module and ScriptModule.

static add_argparse_arguments(parser: ArgumentParser) → None

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 !

Parameters:

parser (ArgumentParser) – the argument parser