Skip to content

Commit

Permalink
added detail about the first fully-connected layer
Browse files Browse the repository at this point in the history
  • Loading branch information
DJTobias committed Feb 5, 2017
1 parent b025aed commit 13e21bc
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion Tensorflow/car_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,10 @@
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
"""

"""
Original net from the paper had 1164 n_neurons on the first fully-connected layer instead of 512.
Halving the amount of neurons on the first FC layer reduced the size of trained model by half and sped up training.
"""

import tensorflow as tf
import numpy as np
Expand Down

0 comments on commit 13e21bc

Please sign in to comment.