tf.nn.conv2d vs tf.layers.conv2d

by

As GBY mentioned, they use the same implementation.

There is a slight difference in the parameters.

For tf.nn.conv2d:

filter: A Tensor. Must have the same type as input. A 4-D tensor of shape [filter_height, filter_width, in_channels, out_channels]

For tf.layers.conv2d:

filters: Integer, the dimensionality of the output space (i.e. the number of filters in the convolution).

I would use tf.nn.conv2d when loading a pretrained model (example code: https://github.com/ry/tensorflow-vgg16), and tf.layers.conv2d for a model trained from scratch.

More Related Contents:

  1. How to save/restore a model after training?
  2. How to prevent tensorflow from allocating the totality of a GPU memory?
  3. Keras, How to get the output of each layer?
  4. Deep-Learning Nan loss reasons
  5. Failed to get convolution algorithm. This is probably because cuDNN failed to initialize,
  6. Why is TensorFlow 2 much slower than TensorFlow 1?
  7. Tensorflow: How to replace or modify gradient?
  8. Fix not load dynamic library for Tensorflow GPU
  9. Can Keras with Tensorflow backend be forced to use CPU or GPU at will?
  10. ImportError: No module named tensorflow
  11. Cannot import Tensorflow for GPU on Windows 10
  12. What is the difference between tf.keras and tf.python.keras?
  13. TensorFlow: Remember LSTM state for next batch (stateful LSTM)
  14. Tensorflow set CUDA_VISIBLE_DEVICES within jupyter
  15. How to *actually* read CSV data in TensorFlow?
  16. Negative dimension size caused by subtracting 3 from 1 for ‘conv2d_2/convolution’
  17. Dimension of shape in conv1D
  18. Compute pairwise distance in a batch without replicating tensor in Tensorflow?
  19. Replacing placeholder for tensorflow v2
  20. How to know if underfitting or overfitting is occuring?
  21. Is there a simpler way to handle batch inputs from tfrecords?
  22. Loading a trained Keras model and continue training
  23. Multiple outputs in Keras
  24. Can’t install Tensorflow Mac
  25. How to implement Grad-CAM on a trained network
  26. How to fix “AttributeError: module ‘tensorflow’ has no attribute ‘get_default_graph'”?
  27. What is right batch normalization function in Tensorflow?
  28. Is it thread-safe when using tf.Session in inference service?
  29. Strange behaviour of the loss function in keras model, with pretrained convolutional base
  30. TensorFlow 2.0 Keras: How to write image summaries for TensorBoard

Leave a Comment