search

phlippe / uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
https://uvadlc-notebooks.readthedocs.io/en/latest/
MIT License
2.59k stars 590 forks source link

tutorial 2 (intro to jax): checkpoint API has changed #79

Open murphyk opened 1 year ago

murphyk commented 1 year ago

Tutorial

https://uvadlc-notebooks.readthedocs.io/en/latest/tutorial_notebooks/JAX/tutorial2/Introduction_to_JAX.html

Describe the bug

The checkpoint saving/loading code needs to be changed to the new flax API

https://flax.readthedocs.io/en/latest/guides/orbax_upgrade_guide.html

phlippe commented 1 year ago

Thanks for pointing out the API change! It seems to me that the API is still in progress and possible subject to change, so I'll wait a bit longer before changing it everywhere in the tutorials.