use of com.simiacryptus.mindseye.layers.cudnn.ImgBandBiasLayer in project MindsEye by SimiaCryptus.
the class VGG19_HDF5 method phase3a.
/**
* Phase 3 a.
*/
protected void phase3a() {
add(new ConvolutionLayer(1, 1, 4096, 4096).setPaddingXY(0, 0).setAndFree(hdf5.readDataSet("param_0", "layer_40").permuteDimensionsAndFree(fullyconnectedOrder)));
add(new ImgBandBiasLayer(4096).setAndFree((hdf5.readDataSet("param_1", "layer_40"))));
add(new ActivationLayer(ActivationLayer.Mode.RELU));
add(new ConvolutionLayer(1, 1, 4096, 1000).setPaddingXY(0, 0).setAndFree(hdf5.readDataSet("param_0", "layer_42").permuteDimensionsAndFree(fullyconnectedOrder)));
add(new ImgBandBiasLayer(1000).setAndFree((hdf5.readDataSet("param_1", "layer_42"))));
}
use of com.simiacryptus.mindseye.layers.cudnn.ImgBandBiasLayer in project MindsEye by SimiaCryptus.
the class DeepConvolution method addLayer.
@Override
public void addLayer(@Nonnull final PipelineNetwork network, final int[] in, final int[] out) {
assert in[0] == out[0];
assert in[1] == out[1];
network.add(new ConvolutionLayer(radius, radius, in[2], out[2]).set(i -> random()));
network.add(new ImgBandBiasLayer(out[2]));
network.add(getActivation());
}
Aggregations