Skip to content

Instantly share code, notes, and snippets.

@liweigu
Created August 30, 2018 07:36
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save liweigu/7c33cebca0d7da503e032dee03f476f4 to your computer and use it in GitHub Desktop.
Save liweigu/7c33cebca0d7da503e032dee03f476f4 to your computer and use it in GitHub Desktop.
network:
GraphBuilder graphBuilder = builder.graphBuilder().backpropType(BackpropType.Standard).addInputs("inputs")
.addLayer("cnn1",
new ConvolutionLayer.Builder(new int[] { kernelSize1, kernelSize1 },
new int[] { cnnStride1, cnnStride1 },
new int[] { padding, padding })
.nIn(channels)
.nOut(64)
inputPreProcessors.put("cnn1", new RnnToCnnPreProcessor(501, 501, channels));
feature mask:
double[] inputMask = new double[sampleCount * size];
for (int i = 0; i < 30; i++) {
inputMask[i] = 1;
}
featuresMask = Nd4j.create(inputMask, new int[] { size, sampleCount });
DataSet dataSet = new DataSet(featureData, labelData, featuresMask, labelsMask);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment