Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Fix MNIST classification example (#19072)
Browse files Browse the repository at this point in the history
Co-authored-by: r3stl355 <ulmasov@amazon.com>
  • Loading branch information
r3stl355 and ulmasov committed Sep 2, 2020
1 parent 5122d32 commit 4cfb9e1
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 8 deletions.
1 change: 1 addition & 0 deletions CONTRIBUTORS.md
Original file line number Diff line number Diff line change
Expand Up @@ -284,6 +284,7 @@ List of Contributors
* [Connor Goggins](https://github.com/connorgoggins)
* [Wei Chu](https://github.com/waytrue17)
* [Joe Evans](https://github.com/josephevans)
* [Nikolay Ulmasov](https://github.com/r3stl355)

Label Bot
---------
Expand Down
7 changes: 3 additions & 4 deletions example/gluon/mnist/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,9 @@ This script shows a simple example how to do image classification with Gluon.
The model is trained on MNIST digits image dataset and the goal is to classify the digits ```0-9```. The model has the following layout:
```
net = nn.Sequential()
with net.name_scope():
net.add(nn.Dense(128, activation='relu'))
net.add(nn.Dense(64, activation='relu'))
net.add(nn.Dense(10))
net.add(nn.Dense(128, activation='relu'))
net.add(nn.Dense(64, activation='relu'))
net.add(nn.Dense(10))
```

The script provides the following commandline arguments:
Expand Down
7 changes: 3 additions & 4 deletions example/gluon/mnist/mnist.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,10 +48,9 @@
# define network

net = nn.Sequential()
with net.name_scope():
net.add(nn.Dense(128, activation='relu'))
net.add(nn.Dense(64, activation='relu'))
net.add(nn.Dense(10))
net.add(nn.Dense(128, activation='relu'))
net.add(nn.Dense(64, activation='relu'))
net.add(nn.Dense(10))

# data

Expand Down

0 comments on commit 4cfb9e1

Please sign in to comment.