Skip to content

Possibly incorrect layer size #6

@slvrfn

Description

@slvrfn

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"?

nn.Linear(192 * 7 * 7, 3072),

x = x.view(x.size(0), 192 * 7 * 7)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions