Return CNNTrain2Caffe2 in Backend in case of CAFFE2

parent d06ebd76
Pipeline #72955 passed with stage
in 1 minute and 59 seconds
......@@ -18,7 +18,7 @@
<emadl.version>0.2.2-SNAPSHOT</emadl.version>
<CNNTrain.version>0.2.5-SNAPSHOT</CNNTrain.version>
<cnnarch-mxnet-generator.version>0.2.5-SNAPSHOT</cnnarch-mxnet-generator.version>
<cnnarch-caffe2-generator.version>0.2.2-SNAPSHOT</cnnarch-caffe2-generator.version>
<cnnarch-caffe2-generator.version>0.2.5-SNAPSHOT</cnnarch-caffe2-generator.version>
<embedded-montiarc-math-generator>0.0.25-SNAPSHOT</embedded-montiarc-math-generator>
<!-- .. Libraries .................................................. -->
......
......@@ -5,6 +5,7 @@ import de.monticore.lang.monticar.cnnarch.CNNArchGenerator;
import de.monticore.lang.monticar.cnnarch.mxnetgenerator.CNNArch2MxNet;
import de.monticore.lang.monticar.cnnarch.caffe2generator.CNNArch2Caffe2;
import de.monticore.lang.monticar.cnnarch.mxnetgenerator.CNNTrain2MxNet;
import de.monticore.lang.monticar.cnnarch.caffe2generator.CNNTrain2Caffe2;
import de.monticore.lang.monticar.cnntrain.CNNTrainGenerator;
import java.util.Optional;
......@@ -27,8 +28,8 @@ public enum Backend {
}
@Override
public CNNTrainGenerator getCNNTrainGenerator() {
return null;
} // not implemented yet
return new CNNTrain2Caffe2();
}
};
public abstract CNNArchGenerator getCNNArchGenerator();
......
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