jaxline

jaxline copied to clipboard

Can you provide some demos on how to use experiment?

3

Can you provide some demos on how to use experiment?

cfangplus

Hi, I am a bit confused about the decide selection if running with `train_eval_multithreaded`. In my set up I have two GPUs and in training I use `pmap` and for...

sigeisler

Hi, I was trying jaxline with 2 local GPUs, but I receive ` RESOURCE_EXHAUSTED: Out of memory while trying to allocate 285212672 bytes.` errors. I have sufficient memory available on...

devavratTomar

How to access InMemory checkpoints and save to disk

2

Please kindly guide or advise on how to access InMemory checkpoints created by the InMemeoryCheckpointer, in order to save the checkpoints to disk. Thank you.

Alikerin

platform flags

3

I didn't have problems using debuggers when the form was: `FLAGS = flags.FLAGS` and later on: ``` # Make sure the required fields are available in the config. base_config.validate_config(FLAGS.config) ```...

jameschapman19

Hi, I noticed that an Experiment's `snapshot_state ` is not invoked anywhere, which seems to imply that the `InMemoryCheckpointer` is not really checkpointing anything? Am I missing something? Thanks

thisiscam