contango00 issues

Repositories
Issues
Comments

Results 1 issues of


                                            contango00

jax.devices issue with the finetuning notebook

is the versioning correct in the yml? JAX 4.26? with the DEVICES = np.array(jax.devices()).reshape([1, 1, 1]) statement in the finetuning notebook I am getting the following error ValueError: cannot reshape...