Commit d4f356a9 authored by Sebastian Nickels's avatar Sebastian Nickels

Small changes to the checkers in CNNArch2MxNet class

parent 5f326fc2
Pipeline #150610 failed with stages
......@@ -36,13 +36,18 @@ import java.util.Map;
public class CNNArch2MxNet extends CNNArchGenerator {
protected ArchitectureSupportChecker architectureSupportChecker;
protected LayerSupportChecker layerSupportChecker;
public CNNArch2MxNet() {
architectureSupportChecker = new CNNArch2MxNetArchitectureSupportChecker();
layerSupportChecker = new CNNArch2MxNetLayerSupportChecker();
setGenerationTargetPath("./target/generated-sources-cnnarch/");
}
public void generate(Scope scope, String rootModelName){
CNNArchSymbolCompiler symbolCompiler = new CNNArchSymbolCompiler(new CNNArch2MxNetArchitectureSupportChecker(),
new CNNArch2MxNetLayerSupportChecker());
CNNArchSymbolCompiler symbolCompiler = new CNNArchSymbolCompiler(architectureSupportChecker, layerSupportChecker);
ArchitectureSymbol architectureSymbol = symbolCompiler.compileArchitectureSymbol(scope, rootModelName);
try{
......
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