Skip to content

Commit

Permalink
Add README for MXNet LeNet example (intel-analytics#2208)
Browse files Browse the repository at this point in the history
* initial readme

* update

* typo

* typo

* add batch size

* update style

* add validation for gluon

* fix style

* minor

* update

* minor

* more doc

* minor
  • Loading branch information
hkvision committed Apr 14, 2020
1 parent 845d78a commit 23298f2
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion finetune/image_finetuning_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
parser.add_option("--b", "--batch_size", type=int, dest="batch_size", default="56",
help="The number of samples per gradient update. Default is 56.")
parser.add_option("--nb_epoch", type=int, dest="nb_epoch", default="2",
help="The number of iterations to train the model. Default is 2.")
help="The number of epochs to train the model. Default is 2.")
parser.add_option("--r", "--learning_rate", type=float, dest="learning_rate", default="0.003",
help="The learning rate for the model. Default is 0.003.")

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@
parser.add_option("--b", "--batch_size", type=int, dest="batch_size", default="56",
help="The number of samples per gradient update. Default is 56.")
parser.add_option("--nb_epoch", type=int, dest="nb_epoch", default="20",
help="The number of iterations to train the model. Default is 20.")
help="The number of epochs to train the model. Default is 20.")
parser.add_option("--r", "--learning_rate", type=float, dest="learning_rate", default="0.002",
help="The learning rate for the model. Default is 0.002.")

Expand Down

0 comments on commit 23298f2

Please sign in to comment.