Commit 79890c82 authored by Sebastian Nickels's avatar Sebastian Nickels

Fixed bug

parent 8c800036
...@@ -171,7 +171,7 @@ public abstract class CNNArchTemplateController { ...@@ -171,7 +171,7 @@ public abstract class CNNArchTemplateController {
return list; return list;
} }
public List getArchitectureInputSymbols(){ public List<VariableSymbol> getArchitectureInputSymbols(){
Set<String> names = new HashSet(); Set<String> names = new HashSet();
List<VariableSymbol> noDuplicates = new ArrayList(); List<VariableSymbol> noDuplicates = new ArrayList();
...@@ -186,7 +186,7 @@ public abstract class CNNArchTemplateController { ...@@ -186,7 +186,7 @@ public abstract class CNNArchTemplateController {
return noDuplicates; return noDuplicates;
} }
public List getArchitectureOutputSymbols(){ public List<VariableSymbol> getArchitectureOutputSymbols(){
Set<String> names = new HashSet(); Set<String> names = new HashSet();
List<VariableSymbol> noDuplicates = new ArrayList(); List<VariableSymbol> noDuplicates = new ArrayList();
......
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