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