parent
1c39dbee52
commit
345f55a003
|
@ -331,7 +331,6 @@ public abstract class NeuralNetBaseBuilderConfiguration implements INeuralNetwor
|
||||||
*/
|
*/
|
||||||
@Getter @Setter @lombok.Builder.Default private IUpdater biasUpdater = null;
|
@Getter @Setter @lombok.Builder.Default private IUpdater biasUpdater = null;
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Weight initialization scheme to use, for initial weight values Note: values set by this method
|
* Weight initialization scheme to use, for initial weight values Note: values set by this method
|
||||||
* will be applied to all applicable layers in the network, unless a different value is explicitly
|
* will be applied to all applicable layers in the network, unless a different value is explicitly
|
||||||
|
|
|
@ -21,6 +21,8 @@
|
||||||
package org.deeplearning4j.nn.conf.layers;
|
package org.deeplearning4j.nn.conf.layers;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
|
import lombok.NoArgsConstructor;
|
||||||
import lombok.experimental.SuperBuilder;
|
import lombok.experimental.SuperBuilder;
|
||||||
import net.brutex.ai.dnn.api.LayerType;
|
import net.brutex.ai.dnn.api.LayerType;
|
||||||
import org.deeplearning4j.nn.api.ParamInitializer;
|
import org.deeplearning4j.nn.api.ParamInitializer;
|
||||||
|
@ -29,7 +31,7 @@ import org.deeplearning4j.nn.conf.inputs.InputType;
|
||||||
import org.deeplearning4j.nn.params.EmptyParamInitializer;
|
import org.deeplearning4j.nn.params.EmptyParamInitializer;
|
||||||
import org.nd4j.linalg.learning.config.IUpdater;
|
import org.nd4j.linalg.learning.config.IUpdater;
|
||||||
import org.nd4j.linalg.learning.regularization.Regularization;
|
import org.nd4j.linalg.learning.regularization.Regularization;
|
||||||
|
@NoArgsConstructor
|
||||||
@SuperBuilder
|
@SuperBuilder
|
||||||
public abstract class NoParamLayer extends LayerConfiguration {
|
public abstract class NoParamLayer extends LayerConfiguration {
|
||||||
|
|
||||||
|
|
|
@ -40,6 +40,7 @@ import java.util.Collection;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
|
||||||
@Data
|
@Data
|
||||||
|
@NoArgsConstructor
|
||||||
@EqualsAndHashCode(callSuper = true)
|
@EqualsAndHashCode(callSuper = true)
|
||||||
@SuperBuilder(builderMethodName = "innerBuilder")
|
@SuperBuilder(builderMethodName = "innerBuilder")
|
||||||
public class Cropping3D extends NoParamLayer {
|
public class Cropping3D extends NoParamLayer {
|
||||||
|
|
|
@ -827,7 +827,7 @@ public class MultiLayerNetwork extends ArtificialNeuralNetwork
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This method allows you to specificy GradientsAccumulator instance to be used with this model
|
* This method allows you to specify GradientsAccumulator instance to be used with this model
|
||||||
* <br>
|
* <br>
|
||||||
* <br>
|
* <br>
|
||||||
* PLEASE NOTE: Do not use this method unless you understand how to use GradientsAccumulator &
|
* PLEASE NOTE: Do not use this method unless you understand how to use GradientsAccumulator &
|
||||||
|
|
Loading…
Reference in New Issue