Data path is read from architecture symbol

parent ce32aaa3
Pipeline #101402 failed with stages
in 1 minute
......@@ -80,6 +80,7 @@ public class CNNArch2MxNet implements CNNArchGenerator {
CNNArchCocos.checkAll(compilationUnit.get());
try{
compilationUnit.get().getArchitecture().setDataPath("Temporary - read the correct data path from the config!");
generateFiles(compilationUnit.get().getArchitecture());
}
catch (IOException e){
......
......@@ -86,6 +86,10 @@ public class CNNArchTemplateController {
return getArchitecture().getEnclosingScope().getSpanningSymbol().get().getFullName().replaceAll("\\.","_");
}
public String getDataPath(){
return getArchitecture().getDataPath();
}
public List<String> getLayerInputs(ArchitectureElementSymbol layer){
List<String> inputNames = new ArrayList<>();
......
......@@ -18,7 +18,7 @@ class MyConstant(mx.init.Initializer):
class ${tc.fileNameWithoutEnding}:
module = None
_data_dir_ = "data/${tc.fullArchitectureName}/"
_data_dir_ = "data/${tc.dataPath}/"
_model_dir_ = "model/${tc.fullArchitectureName}/"
_model_prefix_ = "${tc.architectureName}"
_input_names_ = [${tc.join(tc.architectureInputs, ",", "'", "'")}]
......
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