jaxline
jaxline copied to clipboard
Can you provide some demos on how to use experiment?
Can you provide some demos on how to use experiment?
FWIW : I came here having seen it being used in their PerceiverIO release : https://github.com/deepmind/deepmind-research/blob/master/perceiver/train/experiment.py
thx, and yea I have also seen another https://github.com/deepmind/deepmind-research/tree/master/ogb_lsc/mag. I think it's not easy to use and understand at that time, so I opened this issue, just hoping the community to provide some demo.
here are some examples to use this repo: https://github.com/deepmind/kfac-jax/tree/main/examples
To run an example simply do:
python example_name/pipeline.py --config=example_name/pipeline.py