Aufgrund einer Wartung wird GitLab am 17.08. zwischen 8:30 und 9:00 Uhr kurzzeitig nicht zur Verfügung stehen. / Due to maintenance, GitLab will be temporarily unavailable on 17.08. between 8:30 and 9:00 am.

Commit 0b46c76e authored by aa's avatar aa Committed by Julian Johannes Steinsberger-Dührßen

Added Tensorflow generator

parent d8a9cb6f
...@@ -68,7 +68,7 @@ public enum Backend { ...@@ -68,7 +68,7 @@ public enum Backend {
case "GLUON": case "GLUON":
return Optional.of(GLUON); return Optional.of(GLUON);
return Optional.of(TENSORFLOW); return Optional.of(TENSORFLOW);
...@@ -81,10 +81,13 @@ public enum Backend { ...@@ -81,10 +81,13 @@ public enum Backend {
switch (backend){ switch (backend){
case CAFFE2: case CAFFE2:
return "CAFFE2"; return "CAFFE2";
case GLUON: case GLUON:
return "GLUON"; return "GLUON";
return "TENSORFLOW"; return "TENSORFLOW";
default: default:
return "MXNET"; return "MXNET";
} }
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