Skip to content
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

merge resnet into master #111

Merged
merged 40 commits into from
Nov 18, 2020
Merged

merge resnet into master #111

merged 40 commits into from
Nov 18, 2020

Commits on Sep 6, 2020

  1. save

    cgarciae committed Sep 6, 2020
    Configuration menu
    Copy the full SHA
    cb39a25 View commit details
    Browse the repository at this point in the history

Commits on Sep 7, 2020

  1. initial refactor

    cgarciae committed Sep 7, 2020
    Configuration menu
    Copy the full SHA
    ac9664e View commit details
    Browse the repository at this point in the history

Commits on Sep 8, 2020

  1. jit

    cgarciae committed Sep 8, 2020
    Configuration menu
    Copy the full SHA
    fbbec1a View commit details
    Browse the repository at this point in the history
  2. jit + init_jit

    cgarciae committed Sep 8, 2020
    Configuration menu
    Copy the full SHA
    2def58f View commit details
    Browse the repository at this point in the history
  3. handle rng

    cgarciae committed Sep 8, 2020
    Configuration menu
    Copy the full SHA
    a1c3c13 View commit details
    Browse the repository at this point in the history

Commits on Sep 10, 2020

  1. jit + value_and_grad

    cgarciae committed Sep 10, 2020
    Configuration menu
    Copy the full SHA
    bc2a313 View commit details
    Browse the repository at this point in the history

Commits on Sep 13, 2020

  1. save

    cgarciae committed Sep 13, 2020
    Configuration menu
    Copy the full SHA
    252167d View commit details
    Browse the repository at this point in the history
  2. save

    cgarciae committed Sep 13, 2020
    Configuration menu
    Copy the full SHA
    7f28551 View commit details
    Browse the repository at this point in the history
  3. save

    cgarciae committed Sep 13, 2020
    Configuration menu
    Copy the full SHA
    79aedd0 View commit details
    Browse the repository at this point in the history

Commits on Sep 14, 2020

  1. fix metrics_loss

    cgarciae committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    1c20b3b View commit details
    Browse the repository at this point in the history
  2. save

    cgarciae committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    2541254 View commit details
    Browse the repository at this point in the history
  3. save

    cgarciae committed Sep 14, 2020
    Configuration menu
    Copy the full SHA
    39aa42a View commit details
    Browse the repository at this point in the history

Commits on Sep 15, 2020

  1. *_on_batch methods

    cgarciae committed Sep 15, 2020
    Configuration menu
    Copy the full SHA
    f575d9c View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2020

  1. get_states

    cgarciae committed Sep 18, 2020
    Configuration menu
    Copy the full SHA
    34ea86e View commit details
    Browse the repository at this point in the history

Commits on Sep 19, 2020

  1. save

    cgarciae committed Sep 19, 2020
    Configuration menu
    Copy the full SHA
    6fdf3ba View commit details
    Browse the repository at this point in the history
  2. fix tests

    cgarciae committed Sep 19, 2020
    Configuration menu
    Copy the full SHA
    b9a6616 View commit details
    Browse the repository at this point in the history
  3. fix examples

    cgarciae committed Sep 19, 2020
    Configuration menu
    Copy the full SHA
    250ec66 View commit details
    Browse the repository at this point in the history
  4. format black

    cgarciae committed Sep 19, 2020
    Configuration menu
    Copy the full SHA
    ef9abe2 View commit details
    Browse the repository at this point in the history
  5. use pickle only to save

    cgarciae committed Sep 19, 2020
    Configuration menu
    Copy the full SHA
    68d71d6 View commit details
    Browse the repository at this point in the history
  6. clean model

    cgarciae committed Sep 19, 2020
    Configuration menu
    Copy the full SHA
    24b4811 View commit details
    Browse the repository at this point in the history

Commits on Sep 26, 2020

  1. save

    cgarciae committed Sep 26, 2020
    Configuration menu
    Copy the full SHA
    f07043f View commit details
    Browse the repository at this point in the history
  2. [Fix] Return all files to 0644 file permisions

    David Cardozo committed Sep 26, 2020
    Configuration menu
    Copy the full SHA
    b7972f3 View commit details
    Browse the repository at this point in the history
  3. fix docs

    cgarciae committed Sep 26, 2020
    Configuration menu
    Copy the full SHA
    28d5a6c View commit details
    Browse the repository at this point in the history

Commits on Oct 3, 2020

  1. update module-system guide

    cgarciae committed Oct 3, 2020
    Configuration menu
    Copy the full SHA
    0431709 View commit details
    Browse the repository at this point in the history
  2. update README

    cgarciae committed Oct 3, 2020
    Configuration menu
    Copy the full SHA
    3869335 View commit details
    Browse the repository at this point in the history

Commits on Oct 6, 2020

  1. fix elegy.jit

    cgarciae committed Oct 6, 2020
    Configuration menu
    Copy the full SHA
    ee8bd19 View commit details
    Browse the repository at this point in the history

Commits on Nov 1, 2020

  1. update jax

    cgarciae committed Nov 1, 2020
    Configuration menu
    Copy the full SHA
    2c55db8 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    ea113d2 View commit details
    Browse the repository at this point in the history
  3. fix tests

    cgarciae committed Nov 1, 2020
    Configuration menu
    Copy the full SHA
    61d4f53 View commit details
    Browse the repository at this point in the history

Commits on Nov 2, 2020

  1. small refactor

    cgarciae committed Nov 2, 2020
    Configuration menu
    Copy the full SHA
    37da814 View commit details
    Browse the repository at this point in the history

Commits on Nov 8, 2020

  1. jupyter dev dependency

    cgarciae committed Nov 8, 2020
    Configuration menu
    Copy the full SHA
    1cfc9e2 View commit details
    Browse the repository at this point in the history
  2. update docs

    cgarciae committed Nov 8, 2020
    Configuration menu
    Copy the full SHA
    c1ead24 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    1bc4dc6 View commit details
    Browse the repository at this point in the history
  4. use --no-hashes

    cgarciae committed Nov 8, 2020
    Configuration menu
    Copy the full SHA
    a107c27 View commit details
    Browse the repository at this point in the history
  5. use --without-hashes

    cgarciae committed Nov 8, 2020
    Configuration menu
    Copy the full SHA
    90f007d View commit details
    Browse the repository at this point in the history
  6. Configuration menu
    Copy the full SHA
    7cfe268 View commit details
    Browse the repository at this point in the history
  7. Configuration menu
    Copy the full SHA
    12950f6 View commit details
    Browse the repository at this point in the history

Commits on Nov 9, 2020

  1. fix wraps init

    cgarciae committed Nov 9, 2020
    Configuration menu
    Copy the full SHA
    f341e2b View commit details
    Browse the repository at this point in the history

Commits on Nov 18, 2020

  1. Resnet (#108)

    * added resnet18
    
    * imagenet input pipeline, from https://github.com/google/flax
    
    * experimental support for mixed precision
    
    * full training script
    
    * black + resnet test
    
    * format black
    
    * re-jit when loading a model for compability among platforms
    
    * format black
    
    * use different poetry installer
    
    Co-authored-by: Cristian Garcia <cgarcia.e88@gmail.com>
    alexander-g and cgarciae committed Nov 18, 2020
    Configuration menu
    Copy the full SHA
    22757e0 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    565efca View commit details
    Browse the repository at this point in the history