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:
@@ -2,3 +2,4 @@ datasets >= 1.1.3
|
|||||||
jax>=0.2.8
|
jax>=0.2.8
|
||||||
jaxlib>=0.1.59
|
jaxlib>=0.1.59
|
||||||
flax>=0.3.4
|
flax>=0.3.4
|
||||||
|
optax>=0.0.8
|
||||||
|
|||||||
@@ -2,4 +2,4 @@ datasets >= 1.1.3
|
|||||||
jax>=0.2.8
|
jax>=0.2.8
|
||||||
jaxlib>=0.1.59
|
jaxlib>=0.1.59
|
||||||
flax>=0.3.4
|
flax>=0.3.4
|
||||||
git+https://github.com/deepmind/optax.git
|
optax>=0.0.8
|
||||||
|
|||||||
Reference in New Issue
Block a user