Commit 349d487d authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Changed some layers to trainable

parent 54ec67ae
Pipeline #200228 passed with stages
in 17 minutes and 52 seconds
...@@ -34,11 +34,6 @@ public class OneHot extends PredefinedLayerDeclaration { ...@@ -34,11 +34,6 @@ public class OneHot extends PredefinedLayerDeclaration {
super(AllPredefinedLayers.ONE_HOT_NAME); super(AllPredefinedLayers.ONE_HOT_NAME);
} }
@Override
public boolean isTrainable() {
return false;
}
@Override @Override
public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) { public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) {
return Collections.singletonList(new ArchTypeSymbol.Builder() return Collections.singletonList(new ArchTypeSymbol.Builder()
......
...@@ -34,11 +34,6 @@ public class Split extends PredefinedLayerDeclaration { ...@@ -34,11 +34,6 @@ public class Split extends PredefinedLayerDeclaration {
super(AllPredefinedLayers.SPLIT_NAME); super(AllPredefinedLayers.SPLIT_NAME);
} }
@Override
public boolean isTrainable() {
return false;
}
@Override @Override
public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) { public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) {
ArchTypeSymbol inputShape = layer.getInputTypes().get(0); ArchTypeSymbol inputShape = layer.getInputTypes().get(0);
......
...@@ -35,11 +35,6 @@ public class SwapAxes extends PredefinedLayerDeclaration { ...@@ -35,11 +35,6 @@ public class SwapAxes extends PredefinedLayerDeclaration {
super(AllPredefinedLayers.SWAPAXES_NAME); super(AllPredefinedLayers.SWAPAXES_NAME);
} }
@Override
public boolean isTrainable() {
return false;
}
@Override @Override
public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) { public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) {
......
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