Skip to content
This repository has been archived by the owner on Feb 26, 2023. It is now read-only.

Bumps flax to 0.4.0 #60

Merged
merged 3 commits into from Feb 4, 2022
Merged

Bumps flax to 0.4.0 #60

merged 3 commits into from Feb 4, 2022

Commits on Feb 4, 2022

  1. Bumps flax to 0.4.0 

    Updates flax to the most recent version. This currently breaks the
current implementation and way in which rng keys of  dropout is being
handled.

Currently have disabled one of the dropout equivalence tests as I am not
fully aware if there is a method of directly affecting the value of
`next_key` within a treex module.
    @ptigwe
    ptigwe committed Feb 4, 2022
    Configuration menu
    Copy the full SHA
    65b511c View commit details
    Browse the repository at this point in the history

  2. [auto] black linting fixes

    @ptigwe
    ptigwe committed Feb 4, 2022
    Configuration menu
    Copy the full SHA
    56cd818 View commit details
    Browse the repository at this point in the history

  3. fix test_pretrained_flax_module

    @cgarciae
    cgarciae committed Feb 4, 2022
    Configuration menu
    Copy the full SHA
    5d2f9c3 View commit details
    Browse the repository at this point in the history