Utils_net-checkpoint

Module Contents

Functions

build_multimodal_linear_encoder()
build_multimodal_linear_decoder()
build_conv_layer(in_channels, out_channels[, flatten, ...])
build_conv_layer_reverse(in_channels, out_channels[, ...])
build_sequential_layer(dim1, dim2[, linearize])
build_last_layer_from_output_dist(dim, input_size, ...)
build_last_layer_from_output_dist_flat(dim, ...)
build_last_layer_from_output_dist_conv(dim, ...[, ...])
build_layer_general(in_channels, out_channels, nn_type)
calculate_flattened_size(image_size, num_conv_layers, ...)
calculate_flattened_size_decoder(image_size, ...)

Utils_net-checkpoint.build_multimodal_linear_encoder()

Utils_net-checkpoint.build_multimodal_linear_decoder()

Utils_net-checkpoint.build_conv_layer(in_channels, out_channels, flatten=False, kernel_size=3, stride=1, padding=1, pool_size=2, pool_stride=2)

Utils_net-checkpoint.build_conv_layer_reverse(in_channels, out_channels, flatten=False, kernel_size=3, stride=1, padding=1, pool_size=2, pool_stride=2)

Utils_net-checkpoint.build_sequential_layer(dim1, dim2, linearize=False)

Utils_net-checkpoint.build_last_layer_from_output_dist(dim, input_size, output_type, nn_type, **kargs)

Utils_net-checkpoint.build_last_layer_from_output_dist_flat(dim, input_size, output_type)

Utils_net-checkpoint.build_last_layer_from_output_dist_conv(dim, input_size, output_type, kernel_size=3, stride=1, padding=1)

Utils_net-checkpoint.build_layer_general(in_channels, out_channels, nn_type, linearize=False, flatten=False, **kargs)

Utils_net-checkpoint.calculate_flattened_size(image_size, num_conv_layers, D, filter_size, stride, padding, pool_size, pool_stride)

Utils_net-checkpoint.calculate_flattened_size_decoder(image_size, num_conv_layers, D, filter_size, stride, padding, pool_stride)