Giter Club home page Giter Club logo

naivenasflux.jl's Introduction

NaiveNASflux

Build status Build Status Codecov

NaiveNASflux uses NaiveNASlib to enable mutation operations of arbitrary Flux computation graphs. It is designed with Neural Architecture Search (NAS) in mind, but can be used for any purpose where doing changes to a model is desired.

Note that NaiveNASflux does not have any functionality to search for a model architecture. Check out NaiveGAflux for a simple proof of concept.

Basic Usage

]add NaiveNASflux

See documentation for usage instructions.

Contributing

All contributions are welcome. Please file an issue before creating a PR.

naivenasflux.jl's People

Contributors

drchainsaw avatar github-actions[bot] avatar juliatagbot avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

naivenasflux.jl's Issues

TagBot trigger issue

This issue is used to trigger TagBot; feel free to unsubscribe.

If you haven't already, you should update your TagBot.yml to include issue comment triggers.
Please see this post on Discourse for instructions and more details.

If you'd like for me to do this for you, comment TagBot fix on this issue.
I'll open a PR within a few hours, please be patient!

ActivationContribution performance bottleneck

Seems to happen for large activations (typically 2D conv with 500+ feature maps). The symptoms are low GPU utilization which seems to be due to excessive GC:ing.

The elementwise multiplication seems to be the culprit as replacing everything inside the gradient hook with just act .* grad gives more or less identical performance as the full implementation.

Don't think there is much I can do about the operation itself, but given that the metric is a bit of a statistical thing it is probably ok to enable computing it

  1. every X batches
  2. on a subset of the examples in a batch

Need to do a few more experiments to see which one seems more useful and/or if both shall be allowed.

Checklist for 2.0

Rel 2 of NaiveNASlib will force a rel 2 due to reexport. Just as with NaiveNASlib, I plan to be quite liberal with breakage. Let me know if you are a user and would prefer less breakage.

Other things to fix:

  • Rename mutable to fluxvertex (wtf was I thinking?!).
  • Don't reexport Flux.
  • Greatly reduce the amount of exported functions and types.
  • Merge #53.
  • Make all layerwrappers real functors instead of relying on mutation (if possible).
  • Rename neuron_value to neuronutility (same for types affecting its behaviour)

Memory leak with stateful optimizers

Stateful optimizers (like ADAM) store state per parameter array in a dict. Since parameter arrays are replaced with new instances when mutating the dict just keeps getting new entries.

This also obviously prevents the optimizers from working as they should.

Support for Conv with groups (> 1) != nin

Current implementation is from the days when DepthwiseConv was the only grouped convolution in Flux. DepthwiseConv was hardcoded to have groups == nin.

I think/hope that constraints can be handled by just applying the current constrains to subsets of the in/out variables. Not sure this will enable the full flexibilty of changing the nin/groups ratio but I'm also not sure if it is meaningful to allow more flexibilty than what the current constraints allow.

What could be the biggest headache is to let the part which creates the new parameter arrays to understand the number of groups and/or the groups/nin ratio from just the selected in/out indicies. Perhaps one needs to be able to pass arbitrary other data in the result from the solver which would require updates to NaiveNASlib.

Will save this one for a rainy day...

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.