Mingfei

@chapulina Hi Louise, if this issue has not been addressed yet, I'd like to work out a solution for it.

You can find out the corresponding parameter positions in your network by checking the following part at the beginning of your training: > Resulting replaced keras model: > `1: def...

Hi RudyAramayo, I came cross the same problem with ZR300. Did you solved it? Thanks

Hi chataign, I also come across this problem. And still not figure it out. Have you solved it?

Indeed, I reinstalled the system to Ubuntu 16.04 and everything works right now. Thank you so much for your advice.

This is likely caused by the python versions. Upgrading to python3.8+ can solve the issue.

here are some examples to use this repo: https://github.com/deepmind/kfac-jax/tree/main/examples To run an example simply do: ``` bash python example_name/pipeline.py --config=example_name/pipeline.py ```