DigitMagazine

Layer creation for neural network in Tensorflow

Jul 25th, 2018
77
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.06 KB | None | 0 0
  1.  
  2. layer_conv1 = create_convolutional_layer(input=x,
  3. num_input_channels=num_channels,
  4. conv_filter_size=filter_size_conv1,
  5. num_filters=num_filters_conv1)
  6. layer_conv2 = create_convolutional_layer(input=layer_conv1,
  7. num_input_channels=num_filters_conv1,
  8. conv_filter_size=filter_size_conv2,
  9. num_filters=num_filters_conv2)
  10.  
  11. layer_conv3= create_convolutional_layer(input=layer_conv2,
  12. num_input_channels=num_filters_conv2,
  13. conv_filter_size=filter_size_conv3,
  14. num_filters=num_filters_conv3)
  15.  
  16. layer_flat = create_flatten_layer(layer_conv3)
  17.  
  18. layer_fc1 = create_fc_layer(input=layer_flat,
  19. num_inputs=layer_flat.get_shape()[1:4].num_elements(),
  20. num_outputs=fc_layer_size,
  21. use_relu=True)
  22.  
  23. layer_fc2 = create_fc_layer(input=layer_fc1,
  24. num_inputs=fc_layer_size,
  25. num_outputs=num_classes,
  26. use_relu=False)
Add Comment
Please, Sign In to add comment