Commit 39c80bc0 authored by Julian Johannes Steinsberger-Dührßen's avatar Julian Johannes Steinsberger-Dührßen
Browse files


parent d7018253
Pipeline #344506 passed with stage
in 1 minute and 24 seconds
......@@ -196,7 +196,13 @@ public class ArchitectureElementData {
public List<Integer> getOutputShape(){
return getLayerSymbol().getIntTupleValue(AllPredefinedLayers.OUTPUT_SHAPE_NAME).get();
if (getLayerSymbol().getIntValue(AllPredefinedLayers.OUTPUT_SHAPE_NAME).isPresent()){
List<Integer> list = new ArrayList<>();
list.add((Integer) getLayerSymbol().getIntValue(AllPredefinedLayers.OUTPUT_SHAPE_NAME).get());
return list;
return getLayerSymbol().getIntTupleValue(AllPredefinedLayers.OUTPUT_SHAPE_NAME).get();
public int getScaleFactor(){
......@@ -52,8 +52,6 @@ public abstract class TrainParamSupportChecker implements CNNTrainVisitor {
public void visit(ASTAdamOptimizer node){}
public void visit(ASTAdamWOptimizer node){}
public void visit(ASTRmsPropOptimizer node){}
public void visit(ASTAdaGradOptimizer node){}
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