sconce.models.layers package

sconce.models.layers.convolution2d_layer module

class sconce.models.layers.convolution2d_layer.Convolution2dLayer(*, in_channels, out_channels, stride=2, kernel_size=3, padding=1, activation=ReLU(), preactivate=False, with_batchnorm=True)[source]

Bases: torch.nn.modules.module.Module

forward(x)[source]
out_height(in_height)[source]
out_size(in_size)[source]
out_width(in_width)[source]

sconce.models.layers.fully_connected_layer module

class sconce.models.layers.fully_connected_layer.FullyConnectedLayer(in_size, out_size, activation=ReLU(), with_batchnorm=True, dropout=0.0)[source]

Bases: torch.nn.modules.module.Module

forward(x_in)[source]