Merge remote-tracking branch 'eclipse/master'
commit
9b3576bc00
|
@ -334,6 +334,7 @@ public class KerasLayerConfiguration {
|
|||
private final String KERAS_ACTIVATION_SIGMOID = "sigmoid";
|
||||
private final String KERAS_ACTIVATION_HARD_SIGMOID = "hard_sigmoid";
|
||||
private final String KERAS_ACTIVATION_LINEAR = "linear";
|
||||
private final String KERAS_ACTIVATION_SWISH = "swish";
|
||||
private final String KERAS_ACTIVATION_ELU = "elu"; // keras 2 only
|
||||
private final String KERAS_ACTIVATION_SELU = "selu"; // keras 2 only
|
||||
|
||||
|
|
|
@ -63,6 +63,8 @@ public class KerasActivationUtils {
|
|||
dl4jActivation = Activation.HARDSIGMOID;
|
||||
} else if (kerasActivation.equals(conf.getKERAS_ACTIVATION_LINEAR())) {
|
||||
dl4jActivation = Activation.IDENTITY;
|
||||
} else if (kerasActivation.equals(conf.getKERAS_ACTIVATION_SWISH())) {
|
||||
dl4jActivation = Activation.SWISH;
|
||||
} else {
|
||||
throw new UnsupportedKerasConfigurationException(
|
||||
"Unknown Keras activation function " + kerasActivation);
|
||||
|
|
|
@ -230,8 +230,8 @@
|
|||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<configuration>
|
||||
<source>8</source>
|
||||
<target>8</target>
|
||||
<source>7</source>
|
||||
<target>7</target>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
|
|
Loading…
Reference in New Issue