Commit ff126d0a authored by Evgeny Kusmenko's avatar Evgeny Kusmenko

Update CNNArch2Caffe2.java

parent 5646c9ab
Pipeline #106582 passed with stages
in 3 minutes and 29 seconds
......@@ -51,6 +51,7 @@ public class CNNArch2Caffe2 implements CNNArchGenerator{
private boolean isSupportedLayer(ArchitectureElementSymbol element, LayerSupportChecker layerChecker){
List<ArchitectureElementSymbol> constructLayerElemList;
//first part of if condition ensures that element is not an IOSymbol
if (element instanceof LayerSymbol && (element.getResolvedThis().get() instanceof CompositeElementSymbol)) {
constructLayerElemList = ((CompositeElementSymbol)element.getResolvedThis().get()).getElements();
for (ArchitectureElementSymbol constructedLayerElement : constructLayerElemList) {
......
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