flax
f384b331 - correct argnum indexes for transforms.checkpoint

Commit

3 years ago

correct argnum indexes for transforms.checkpoint

References

#2457 - Add static_argnums to nn.checkpoint

Author

cgarciae

cgarciae

Parents

Loading