Fixing tests

Signed-off-by: brian <brian@brutex.de>
enhance-build-infrastructure
Brian Rosenberger 2023-05-08 12:48:42 +02:00
parent 1c39dbee52
commit 345f55a003
4 changed files with 5 additions and 3 deletions

View File

@ -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

View File

@ -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 {

View File

@ -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 {

View File

@ -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 &