Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- ZooModel zooModel = new VGG16();
- System.out.println("Start Downloading VGG16 model...");
- ComputationGraph preTrainedNet = (ComputationGraph) zooModel.initPretrained(PretrainedType.IMAGENET);
- System.out.println("Init summary...");
- System.out.println(preTrainedNet.summary());
- System.out.println("END summary...");
- //output...
- SLF4J: Failed to load class "org.slf4j.impl.StaticLoggerBinder".
- SLF4J: Defaulting to no-operation (NOP) logger implementation
- SLF4J: See http://www.slf4j.org/codes.html#StaticLoggerBinder for further details.
- Start Downloading VGG16 model...
- Init summary...
- ============================================================================================================================================
- VertexName (VertexType) nIn,nOut TotalParams ParamsShape Vertex Inputs
- ============================================================================================================================================
- input_1 (InputVertex) -,- - - -
- block1_conv1 (ConvolutionLayer) 3,64 1792 b:{1,64}, W:{64,3,3,3} [input_1]
- block1_conv2 (ConvolutionLayer) 64,64 36928 b:{1,64}, W:{64,64,3,3} [block1_conv1]
- block1_pool (SubsamplingLayer) -,- 0 - [block1_conv2]
- block2_conv1 (ConvolutionLayer) 64,128 73856 b:{1,128}, W:{128,64,3,3} [block1_pool]
- block2_conv2 (ConvolutionLayer) 128,128 147584 b:{1,128}, W:{128,128,3,3} [block2_conv1]
- block2_pool (SubsamplingLayer) -,- 0 - [block2_conv2]
- block3_conv1 (ConvolutionLayer) 128,256 295168 b:{1,256}, W:{256,128,3,3} [block2_pool]
- block3_conv2 (ConvolutionLayer) 256,256 590080 b:{1,256}, W:{256,256,3,3} [block3_conv1]
- block3_conv3 (ConvolutionLayer) 256,256 590080 b:{1,256}, W:{256,256,3,3} [block3_conv2]
- block3_pool (SubsamplingLayer) -,- 0 - [block3_conv3]
- block4_conv1 (ConvolutionLayer) 256,512 1180160 b:{1,512}, W:{512,256,3,3} [block3_pool]
- block4_conv2 (ConvolutionLayer) 512,512 2359808 b:{1,512}, W:{512,512,3,3} [block4_conv1]
- block4_conv3 (ConvolutionLayer) 512,512 2359808 b:{1,512}, W:{512,512,3,3} [block4_conv2]
- block4_pool (SubsamplingLayer) -,- 0 - [block4_conv3]
- block5_conv1 (ConvolutionLayer) 512,512 2359808 b:{1,512}, W:{512,512,3,3} [block4_pool]
- block5_conv2 (ConvolutionLayer) 512,512 2359808 b:{1,512}, W:{512,512,3,3} [block5_conv1]
- block5_conv3 (ConvolutionLayer) 512,512 2359808 b:{1,512}, W:{512,512,3,3} [block5_conv2]
- block5_pool (SubsamplingLayer) -,- 0 - [block5_conv3]
- flatten (PreprocessorVertex) -,- - - [block5_pool]
- fc1 (DenseLayer) 25088,4096 102764544 b:{1,4096}, W:{25088,4096} [flatten]
- fc2 (DenseLayer) 4096,4096 16781312 b:{1,4096}, W:{4096,4096} [fc1]
- predictions (DenseLayer) 4096,1000 4097000 b:{1,1000}, W:{4096,1000} [fc2]
- --------------------------------------------------------------------------------------------------------------------------------------------
- Total Parameters: 138357544
- Trainable Parameters: 138357544
- Frozen Parameters: 0
- ============================================================================================================================================
- END summary...
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement