Commit 0402edaf authored by Christopher Jan-Steffen Brix's avatar Christopher Jan-Steffen Brix
Browse files

Store model path to allow its usage in the templates

parent c92c3107
Pipeline #107345 passed with stages
in 19 minutes and 16 seconds
......@@ -34,6 +34,7 @@ import java.util.Map;
public abstract class CNNArchGenerator {
private String generationTargetPath;
private String modelsDirPath;
public static void quitGeneration(){
Log.error("Code generation is aborted");
......@@ -55,7 +56,12 @@ public abstract class CNNArchGenerator {
this.generationTargetPath = generationTargetPath;
}
protected String getModelsDirPath() {
return this.modelsDirPath;
}
public void generate(Path modelsDirPath, String rootModelName){
this.modelsDirPath = modelsDirPath.toString();
final ModelPath mp = new ModelPath(modelsDirPath);
GlobalScope scope = new GlobalScope(mp, new CNNArchLanguage());
generate(scope, rootModelName);
......
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