9237cdaf5b
901: Add option for "Same" padding to conv and pooling layers r=dhairyagandhi96 a=DrChainsaw Fixes #813 This adds the possibility to set "pad=SamePad()" to automatically calculate the amount of padding to apply so that outputsize==inputsize (assuming stide == 1). Comments on API more than welcome. I considered the following options: * Call the type just Same and export it, but I was afraid to cause name collisions due to a too generic name * Call the type Same and not export it * Dispatch on type instead of instance (so that one can type pad=Same instead of pad=Same()) * Supply a method instead of a type, giving a similar API as above. Happy to change to any of the above or to anything else. I don't think that same padding is common for pooling layers, but I added it just for the sake of consistency. It is a separate commit so it can easily be removed if not wanted. Co-authored-by: DrChainsaw <Christian.kyril.skarby@gmail.com> |
||
|---|---|---|
| .github | ||
| docs | ||
| paper | ||
| src | ||
| test | ||
| .gitattributes | ||
| .gitignore | ||
| .gitlab-ci.yml | ||
| .travis.yml | ||
| CITATION.bib | ||
| LICENSE.md | ||
| Manifest.toml | ||
| NEWS.md | ||
| Project.toml | ||
| README.md | ||
| bors.toml | ||
README.md
Flux is an elegant approach to machine learning. It's a 100% pure-Julia stack, and provides lightweight abstractions on top of Julia's native GPU and AD support. Flux makes the easy things easy while remaining fully hackable.
] add Flux
See the documentation or the model zoo for examples.
If you use Flux in your research, please cite our work.