gsnn.gsnn.tests.test_nn

Tests for gsnn.models.NN baseline.

Functions

test_nn_forward_shape()
test_nn_layers_depth()
test_nn_no_norm()

Classes

NN(*args, **kwargs)

Fully-connected baseline: Linear blocks with optional norm, activation, dropout.

gsnn.gsnn.tests.test_nn.test_nn_forward_shape()

gsnn.gsnn.tests.test_nn.test_nn_layers_depth()

gsnn.gsnn.tests.test_nn.test_nn_no_norm()