Use a released version of optax rather than installing from Git. (#12173)

Use a released version of optax rather than installing from Git
This commit is contained in:
Avital Oliver 2021-06-15 13:12:51 +02:00 committed by GitHub
parent 9bc9e59869
commit 9b393240a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions

View File

@ -2,3 +2,4 @@ datasets >= 1.1.3
jax>=0.2.8
jaxlib>=0.1.59
flax>=0.3.4
optax>=0.0.8

View File

@ -2,4 +2,4 @@ datasets >= 1.1.3
jax>=0.2.8
jaxlib>=0.1.59
flax>=0.3.4
git+https://github.com/deepmind/optax.git
optax>=0.0.8