fixes

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;
}else{
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){}
......
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