search

potterhsu / SVHNClassifier-PyTorch

A PyTorch implementation of Multi-digit Number Recognition from Street View Imagery using Deep Convolutional Neural Networks (http://arxiv.org/pdf/1312.6082.pdf)
MIT License
198 stars 45 forks source link

Possibly incorrect layer size #6

Open slvrfn opened 4 years ago

slvrfn commented 4 years ago

After reviewing the mentioned paper in section 5.1:

The fully connected layers contain 3,072 units each

And your other version of this repository written in tensorflow:

https://github.com/potterhsu/SVHNClassifier/blob/475fb68fde1fa1c057cc301ef0af049d9b9c4afb/model.py#L72

Is it possible the following lines were supposed to be "4 4 192"?

https://github.com/potterhsu/SVHNClassifier-PyTorch/blob/5f062e5afc134d9b56ab3dc6282e7762799f5159/model.py#L77

https://github.com/potterhsu/SVHNClassifier-PyTorch/blob/5f062e5afc134d9b56ab3dc6282e7762799f5159/model.py#L102