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!
Parcourir la source

Add test_[one_hot_]input_layer_parameter_count_01().

master
Stanislaw Adaszewski il y a 3 ans
Parent
révision
1b72e41ea9
1 fichiers modifiés avec 13 ajouts et 0 suppressions
  1. +13
    -0
      tests/icosagon/test_input.py

+ 13
- 0
tests/icosagon/test_input.py Voir le fichier

@@ -107,3 +107,16 @@ def test_one_hot_input_layer_04():
layer = OneHotInputLayer(d)
s = repr(layer)
assert s.startswith('Icosagon one-hot input layer')
def test_one_hot_input_layer_parameter_count_01():
d = _some_data()
layer = OneHotInputLayer(d)
assert len(list(layer.parameters())) == 2
def test_input_layer_parameter_count_01():
d = _some_data()
for output_dim in [32, 64, 128]:
layer = InputLayer(d, output_dim)
assert len(list(layer.parameters())) == 2

Chargement…
Annuler
Enregistrer