parent
ac7df3831e
commit
4ad1987a07
|
@ -38,7 +38,7 @@ import com.fasterxml.jackson.databind.module.SimpleModule;
|
|||
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
|
||||
|
||||
@Deprecated
|
||||
@EqualsAndHashCode
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public abstract class BaseEvaluation<T extends BaseEvaluation> extends org.nd4j.evaluation.BaseEvaluation<T> {
|
||||
|
||||
@Getter
|
||||
|
|
|
@ -21,6 +21,7 @@
|
|||
package org.deeplearning4j.nn.conf.layers.objdetect;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.EqualsAndHashCode;
|
||||
import lombok.Getter;
|
||||
import lombok.Setter;
|
||||
import org.deeplearning4j.nn.api.Layer;
|
||||
|
@ -49,6 +50,7 @@ import java.util.List;
|
|||
import java.util.Map;
|
||||
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public class Yolo2OutputLayer extends org.deeplearning4j.nn.conf.layers.Layer {
|
||||
|
||||
private double lambdaCoord;
|
||||
|
|
|
@ -20,10 +20,7 @@
|
|||
|
||||
package org.deeplearning4j.nn.conf.layers.recurrent;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.Getter;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.Setter;
|
||||
import lombok.*;
|
||||
import org.deeplearning4j.nn.api.Layer;
|
||||
import org.deeplearning4j.nn.api.ParamInitializer;
|
||||
import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
|
||||
|
@ -40,6 +37,7 @@ import java.util.Collection;
|
|||
import java.util.Map;
|
||||
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public class SimpleRnn extends BaseRecurrentLayer {
|
||||
|
||||
private boolean hasLayerNorm = false;
|
||||
|
|
|
@ -20,10 +20,7 @@
|
|||
|
||||
package org.deeplearning4j.nn.conf.memory;
|
||||
|
||||
import lombok.AllArgsConstructor;
|
||||
import lombok.Data;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.NonNull;
|
||||
import lombok.*;
|
||||
import org.deeplearning4j.nn.conf.CacheMode;
|
||||
import org.deeplearning4j.nn.conf.inputs.InputType;
|
||||
import org.nd4j.linalg.api.buffer.DataBuffer;
|
||||
|
@ -35,6 +32,7 @@ import java.util.Map;
|
|||
@Data
|
||||
@AllArgsConstructor
|
||||
@NoArgsConstructor
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public class LayerMemoryReport extends MemoryReport {
|
||||
|
||||
private String layerName;
|
||||
|
|
Loading…
Reference in New Issue