Commit acf49043 authored by Sebastian N.'s avatar Sebastian N.

Moved some MXNet-related code to CNNArch2MXNet and CNNArch2Gluon

parent bbf3a334
Pipeline #171052 passed with stages
in 5 minutes and 46 seconds
...@@ -126,18 +126,5 @@ public abstract class CNNArchGenerator { ...@@ -126,18 +126,5 @@ public abstract class CNNArchGenerator {
} }
} }
public Map<String, String> generateCMakeContent(String rootModelName) { public abstract Map<String, String> generateCMakeContent(String rootModelName);
// model name should start with a lower case letter. If it is a component, replace dot . by _ }
rootModelName = rootModelName.replace('.', '_').replace('[', '_').replace(']', '_');
rootModelName = rootModelName.substring(0, 1).toLowerCase() + rootModelName.substring(1);
CMakeConfig cMakeConfig = new CMakeConfig(rootModelName);
cMakeConfig.addModuleDependency(new CMakeFindModule("Armadillo", true));
cMakeConfig.addCMakeCommand("set(LIBS ${LIBS} mxnet)");
Map<String,String> fileContentMap = new HashMap<>();
for (FileContent fileContent : cMakeConfig.generateCMakeFiles()){
fileContentMap.put(fileContent.getFileName(), fileContent.getFileContent());
}
return fileContentMap;
}}
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