-
-
Notifications
You must be signed in to change notification settings - Fork 333
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Simplify LeNet #394
Simplify LeNet #394
Conversation
f37d4f2
to
f419ab4
Compare
vision/conv_mnist/conv_mnist.jl
Outdated
|
||
# Check that it now agrees with the earlier, trained, model: | ||
|
||
lenet2(cpu(x1)) ≈ cpu(lenet(x1)) |
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.
Just repeating the suggestion in #397 (comment) to print some of these out so that users running the script directly aren't completely left without feedback.
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.
See what you think of 9bf6c63, adds more @show
.
It doesn't print many headings etc. when run as a script. Could add more but it's all extra clutter. If you run this as a script, I think you need the source open in another window to learn anything anyway.
There's a lot of junk in these files. Perhaps we should clean them up.
Text can go in README. Code should encourage interactive use, and hence avoid big functions called exactly once. No more parameter structs.
Readable version, since the diff is huge: https://github.com/FluxML/model-zoo/blob/simpler_lenet/vision/conv_mnist/conv_mnist.jl
WIP, would like FluxML/Flux.jl#2186 and JuliaML/MLUtils.jl#144 .Edit: or just FluxML/Flux.jl#2240 now.