Julian Lemmel

icon location Vienna icon location PhD Student @ CPS-Group, TU Wien Biologically plausible Reinforcement Learning

Results 1 issues of Julian Lemmel

TypeError: write(): incompatible function arguments.

2 comment

After upgrading to `jax==0.4.31` I am seeing this error when trying to save a model using the `PyTreeCheckpointer`. Downgrading to `0.4.30` fixed it for now. ``` File ".../site-packages/orbax/checkpoint/checkpointer.py", line 151,...

checkpoint
type:build/install