Skip to content

Instantly share code, notes, and snippets.

@ssadedin
ssadedin / TestConv1D.java
Created August 4, 2020 10:50
Problem with DL4J 1D Convolution
import java.util.ArrayList;
import java.util.List;
import org.deeplearning4j.datasets.iterator.impl.ListDataSetIterator;
import org.deeplearning4j.nn.api.OptimizationAlgorithm;
import org.deeplearning4j.nn.conf.inputs.*;
import org.deeplearning4j.nn.conf.*;
import org.deeplearning4j.nn.conf.layers.*;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
import org.deeplearning4j.nn.weights.WeightInit;