Commit ea6ea14b authored by Luis Rickert's avatar Luis Rickert Committed by lr119628
Browse files

[added]: added check if previous layer is artificial, if yes its name is passed

parent 372d3d3b
...@@ -127,6 +127,9 @@ public abstract class CNNArchTemplateController { ...@@ -127,6 +127,9 @@ public abstract class CNNArchTemplateController {
List<String> inputNames = new ArrayList<>(); List<String> inputNames = new ArrayList<>();
for (ArchitectureElementSymbol input : layer.getPrevious()) { for (ArchitectureElementSymbol input : layer.getPrevious()) {
if(input.isArtificial()){
inputNames.add(getName(input));
}else
if (input.getOutputTypes().size() == 1) { if (input.getOutputTypes().size() == 1) {
inputNames.add(getName(input)); inputNames.add(getName(input));
} else { } else {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment