IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Pārlūkot izejas kodu

Add test_dropout_graph_conv_activation_parameter_count_01().

master
Stanislaw Adaszewski pirms 4 gadiem
vecāks
revīzija
b79487e772
1 mainītis faili ar 8 papildinājumiem un 0 dzēšanām
  1. +8
    -0
      tests/icosagon/test_convolve.py

+ 8
- 0
tests/icosagon/test_convolve.py Parādīt failu

@@ -196,3 +196,11 @@ def test_graph_conv_parameter_count_01():
conv = GraphConv(20, 20, adj_mat)
assert len(list(conv.parameters())) == 1
def test_dropout_graph_conv_activation_parameter_count_01():
adj_mat = torch.rand((10, 20)).round()
conv = DropoutGraphConvActivation(20, 20, adj_mat)
assert len(list(conv.parameters())) == 1

Notiek ielāde…
Atcelt
Saglabāt