gsnn.gsnn.tests.test_node_mlp

Tests for gsnn.models.NodeMLP.

Functions

test_node_mlp_dropout_train_eval()
test_node_mlp_forward_shape()

Classes

NodeMLP(*args, **kwargs)

Small MLP applied independently to each node's channel vector inside a ResBlock.

gsnn.gsnn.tests.test_node_mlp.test_node_mlp_dropout_train_eval()

gsnn.gsnn.tests.test_node_mlp.test_node_mlp_forward_shape()