dgs.utils.nn

Helper and utility methods for creating neural networks using PyTorch.

Module Functions

fc_linear(hidden_layers[, bias, act_func])	Create a Network consisting of one or more fully connected linear layers with input and output sizes given by the hidden_layers.
set_up_hidden_layer_sizes(input_size, ...[, ...])	Given the input and output size of an FC-NN, create a list of the sizes containing each hidden layer in the network.