Commit 24cdb964 authored by lr119628's avatar lr119628
Browse files

[update]: added check for template controller if the architecture contains an AdaNet layer

parent 6eacaeef
......@@ -103,6 +103,10 @@ public abstract class CNNArchTemplateController {
this.nameManager = new LayerNameCreator(architecture);
public boolean containsAdaNet(){
return this.architecture.containsAdaNet();
public String getName(ArchitectureElementSymbol layer){
return nameManager.getName(layer);
Supports Markdown
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