Models

Model starters.

MLP(*fc_sizes)

Creates sequential fully-connected layers FC_1->FC_2->...->FC_N.

MLPDropout(*fc_sizes[, p, p_input])

Creates sequential fully-connected layers FC_1->FC_2->...->FC_N with nn.Dropout in-between.

AutoencoderLinear(*fc_sizes[, p_drop, ...])

The simplest linear AutoEncoder.

SerializableModule(*args, **kwargs)

A serializable module to easily save and restore the attributes, defined in state_attr.