Commit 3d9809dd authored by Evgeny Kusmenko's avatar Evgeny Kusmenko
Browse files

Merge branch 'oneclick_nn_training' into 'master'

Store model path to allow its usage in the templates

See merge request !21
parents c92c3107 1872a256
Pipeline #155663 passed with stages
in 18 minutes and 34 seconds
......@@ -30,7 +30,7 @@
......@@ -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