-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[MXNET-531] MNIST Examples for Scala new API #11250
Conversation
e28cf64
to
2cb18a5
Compare
bfd1634
to
9e8e988
Compare
* See the License for the specific language governing permissions and | ||
* limitations under the License. | ||
*/ | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I would have liked author tag with original author's name as well as yours here since you are just copying existing files and modifying a few lines, but ASF does not with allow author tags with Apache 2.0 License :(. Lets discuss offline.
e9bc664
to
720610e
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @lanking520, small comment to remove commented code in any other PR
@@ -31,6 +31,7 @@ object TrainMnist { | |||
def getMlp: Symbol = { | |||
val data = Symbol.Variable("data") | |||
|
|||
// val fc1 = Symbol.FullyConnected(name = "relu")()(Map("data" -> data, "act_type" -> "relu")) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
no need to keep it commented, you can remove it completely. change it any of your next PRs. I am going to merge it as is instead of removing this line, I don't want the entire CI suite to run for hour.
* Modify MNIST training example to use new Symbol APIs. * Add Tests to the MNIST Train example
Description
This PR contains examples written from the new API.
@nswamy @yzhliu @andrewfayres
All examples pushed here are tested locally on my Mac. It lives in the new package called
org.apache.mxnet.examples
Checklist
Essentials