Skip to content
Snippets Groups Projects
Commit e9cc606a authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 06606e98
No related merge requests found
......@@ -1321,7 +1321,7 @@ class Class1NeuralNetwork(object):
peptide_allele_merge_activation)(current_layer)
densenet_layers = [] if topology == "densenet" else None
for (i, layer_size) in enumerate(layer_size):
for (i, layer_size) in enumerate(layer_sizes):
if densenet_layers is not None:
densenet_layers.append(current_layer)
if len(densenet_layers) > 1:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment