#7225 L2NormalizeVertex equals/hashCode fix

deeplearning4j · Mar 13, 2019 · 8fcbc9a · 8fcbc9a
1 parent 1d996a2
commit 8fcbc9a
Showing 1 changed file with 2 additions and 12 deletions.
diff --git a/...j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/graph/L2NormalizeVertex.java b/...j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/graph/L2NormalizeVertex.java
@@ -17,6 +17,7 @@
 package org.deeplearning4j.nn.conf.graph;
 
 import lombok.Data;
+import lombok.EqualsAndHashCode;
 import lombok.val;
 import org.deeplearning4j.nn.conf.inputs.InputType;
 import org.deeplearning4j.nn.conf.inputs.InvalidInputTypeException;
@@ -37,6 +38,7 @@
  * @author Alex Black (AlexDBlack)
  */
 @Data
+@EqualsAndHashCode(callSuper = false)
 public class L2NormalizeVertex extends GraphVertex {
     public static final double DEFAULT_EPS = 1e-8;
 
@@ -59,18 +61,6 @@ public L2NormalizeVertex clone() {
         return new L2NormalizeVertex(dimension, eps);
     }
 
-    @Override
-    public boolean equals(Object o) {
-        if (!(o instanceof L2NormalizeVertex))
-            return false;
-        return ((L2NormalizeVertex) o).dimension == dimension;
-    }
-
-    @Override
-    public int hashCode() {
-        return 123081189;
-    }
-
     @Override
     public long numParams(boolean backprop) {
         return 0;