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