parent
18e1717a13
commit
1749b9e4af
|
@ -27,7 +27,7 @@ import com.fasterxml.jackson.annotation.JsonProperty;
|
|||
|
||||
@Deprecated
|
||||
@Data
|
||||
@EqualsAndHashCode(exclude = {"auc"}, callSuper = false)
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public class RocCurve extends org.nd4j.evaluation.curves.RocCurve {
|
||||
|
||||
/**
|
||||
|
|
|
@ -36,7 +36,6 @@ import java.util.Arrays;
|
|||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public class LayerVertex extends GraphVertex {
|
||||
|
||||
private NeuralNetConfiguration layerConf;
|
||||
|
|
|
@ -21,6 +21,7 @@
|
|||
package org.deeplearning4j.nn.conf.layers.misc;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.EqualsAndHashCode;
|
||||
import org.deeplearning4j.nn.api.ParamInitializer;
|
||||
import org.deeplearning4j.nn.api.layers.LayerConstraint;
|
||||
import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
|
||||
|
@ -38,6 +39,7 @@ import java.util.Collection;
|
|||
import java.util.List;
|
||||
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public class FrozenLayerWithBackprop extends BaseWrapperLayer {
|
||||
|
||||
public FrozenLayerWithBackprop(@JsonProperty("layer") Layer layer) {
|
||||
|
|
|
@ -21,6 +21,7 @@
|
|||
package org.deeplearning4j.nn.conf.layers.samediff;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.EqualsAndHashCode;
|
||||
import org.deeplearning4j.nn.api.MaskState;
|
||||
import org.deeplearning4j.nn.api.TrainingConfig;
|
||||
import org.deeplearning4j.nn.conf.GradientNormalization;
|
||||
|
@ -44,6 +45,7 @@ import java.util.List;
|
|||
import java.util.Map;
|
||||
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public abstract class SameDiffVertex extends GraphVertex implements TrainingConfig {
|
||||
|
||||
private SDVertexParams vertexParams;
|
||||
|
|
|
@ -20,10 +20,7 @@
|
|||
|
||||
package org.deeplearning4j.nn.conf.layers.util;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.Getter;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.Setter;
|
||||
import lombok.*;
|
||||
import org.deeplearning4j.nn.conf.InputPreProcessor;
|
||||
import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
|
||||
import org.deeplearning4j.nn.conf.inputs.InputType;
|
||||
|
@ -38,6 +35,7 @@ import com.fasterxml.jackson.annotation.JsonProperty;
|
|||
import java.util.Collection;
|
||||
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public class MaskZeroLayer extends BaseWrapperLayer {
|
||||
|
||||
private double maskingValue = 0.0;
|
||||
|
|
|
@ -21,6 +21,7 @@
|
|||
package org.deeplearning4j.nn.conf.layers.wrapper;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.EqualsAndHashCode;
|
||||
import org.deeplearning4j.nn.api.ParamInitializer;
|
||||
import org.deeplearning4j.nn.conf.GradientNormalization;
|
||||
import org.deeplearning4j.nn.conf.InputPreProcessor;
|
||||
|
@ -33,6 +34,7 @@ import org.nd4j.linalg.learning.regularization.Regularization;
|
|||
import java.util.List;
|
||||
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = false)
|
||||
public abstract class BaseWrapperLayer extends Layer {
|
||||
|
||||
protected Layer underlying;
|
||||
|
|
|
@ -61,7 +61,7 @@ public class AeronNDArrayPublisher implements AutoCloseable {
|
|||
public final static int NUM_RETRIES = 100;
|
||||
private boolean compress = true;
|
||||
private static final BusySpinIdleStrategy busySpinIdleStrategy = new BusySpinIdleStrategy();
|
||||
private int publishRetryTimeOut = 3000;
|
||||
@Builder.Default private int publishRetryTimeOut = 3000;
|
||||
|
||||
private void init() {
|
||||
channel = channel == null ? "aeron:udp?endpoint=localhost:40123" : channel;
|
||||
|
|
|
@ -45,7 +45,7 @@ public class AeronNDArraySubscriber implements AutoCloseable {
|
|||
private String channel;
|
||||
// A unique identifier for a stream within a channel. Stream ID 0 is reserved
|
||||
// for internal use and should not be used by applications.
|
||||
private int streamId = -1;
|
||||
@Builder.Default private int streamId = -1;
|
||||
// Maximum number of message fragments to receive during a single 'poll' operation
|
||||
private int fragmentLimitCount;
|
||||
// Create a context, needed for client connection to media driver
|
||||
|
@ -57,7 +57,7 @@ public class AeronNDArraySubscriber implements AutoCloseable {
|
|||
private NDArrayCallback ndArrayCallback;
|
||||
private Aeron aeron;
|
||||
private Subscription subscription;
|
||||
private AtomicBoolean launched = new AtomicBoolean(false);
|
||||
@Builder.Default private AtomicBoolean launched = new AtomicBoolean(false);
|
||||
private Executor executors;
|
||||
|
||||
|
||||
|
|
|
@ -50,7 +50,8 @@ public class NDArrayMessage implements Serializable {
|
|||
private long index;
|
||||
private int[] dimensions;
|
||||
private byte[] chunk;
|
||||
private int numChunks = 0;
|
||||
|
||||
@Builder.Default private int numChunks = 0;
|
||||
//default dimensions: a 1 length array of -1 means use the whole array for an update.
|
||||
private static int[] WHOLE_ARRAY_UPDATE = {-1};
|
||||
//represents the constant for indicating using the whole array for an update (-1)
|
||||
|
|
|
@ -43,17 +43,17 @@ public class AeronNDArrayResponder implements AutoCloseable {
|
|||
private String channel;
|
||||
// A unique identifier for a stream within a channel. Stream ID 0 is reserved
|
||||
// for internal use and should not be used by applications.
|
||||
private int streamId = -1;
|
||||
@Builder.Default private int streamId = -1;
|
||||
// A unique identifier for a stream within a channel. Stream ID 0 is reserved
|
||||
// for internal use and should not be used by applications.
|
||||
private int responseStreamId = -1;
|
||||
@Builder.Default private int responseStreamId = -1;
|
||||
|
||||
// Maximum number of message fragments to receive during a single 'poll' operation
|
||||
private int fragmentLimitCount;
|
||||
// Create a context, needed for client connection to media driver
|
||||
// A separate media driver process need to run prior to running this application
|
||||
private Aeron.Context ctx;
|
||||
private AtomicBoolean running = new AtomicBoolean(true);
|
||||
@Builder.Default private AtomicBoolean running = new AtomicBoolean(true);
|
||||
private final AtomicBoolean init = new AtomicBoolean(false);
|
||||
private static Logger log = LoggerFactory.getLogger(AeronNDArrayResponder.class);
|
||||
private NDArrayHolder ndArrayHolder;
|
||||
|
|
|
@ -42,7 +42,7 @@ public class HostPortPublisher implements AutoCloseable {
|
|||
private int streamId;
|
||||
// The channel (an endpoint identifier) to send the message to
|
||||
private String channel;
|
||||
private boolean init = false;
|
||||
@Builder.Default private boolean init = false;
|
||||
private Aeron.Context ctx;
|
||||
private Aeron aeron;
|
||||
private Publication publication;
|
||||
|
|
Loading…
Reference in New Issue