search

google-research / scenic

Scenic: A Jax Library for Computer Vision Research and Beyond
Apache License 2.0
3.14k stars 417 forks source link

Backward compatibility issue for running Deformable DETR #1079

Open sachinraja13 opened 2 weeks ago

sachinraja13 commented 2 weeks ago

The requirements file for the deformable DETR project uses jax 0.3.17 and flax 0.5.3.

However, I cannot go back in the version beyond 0.4.26 for jax and 0.8.4 for flax since I want to use METAL accelerator which is only compatible with jax 0.4.26 and above.

Because of this, I'm encountering many issues, specifically with the optimizer. It would be very helpful if I can get some help with porting the code to be compatible with newer versions of jax and flax.

Thanks in advance!