|
@@ -188,3 +188,11 @@ def test_dropout_graph_conv_activation_dense_03(): |
|
|
|
|
|
|
|
|
def test_dropout_graph_conv_activation_sparse_03():
|
|
|
def test_dropout_graph_conv_activation_sparse_03():
|
|
|
_test_dropout_graph_conv_activation_03(True)
|
|
|
_test_dropout_graph_conv_activation_03(True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_graph_conv_parameter_count_01():
|
|
|
|
|
|
adj_mat = torch.rand((10, 20)).round()
|
|
|
|
|
|
|
|
|
|
|
|
conv = GraphConv(20, 20, adj_mat)
|
|
|
|
|
|
|
|
|
|
|
|
assert len(list(conv.parameters())) == 1
|