bitorch.models.resnet_e.ResnetE

class bitorch.models.resnet_e.ResnetE(resnete_num_layers: int, input_shape: List[int], num_classes: int = 0)

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.
create	Creates a ResNetE complying to given layer number.

Attributes

name
resnet_spec

__init__(resnete_num_layers: int, input_shape: List[int], num_classes: int = 0) → 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

create(num_layers: int) → Module

Creates a ResNetE complying to given layer number.

Parameters:

num_layers (int) – number of layers to be build.

Raises:

ValueError – raised if no resnet specification for given num_layers is listed in the resnet_spec dict above

Returns:

resnetE model

Return type:

Module