jaxline

jaxline copied to clipboard

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)

But I get errors with the new version:

# Make sure the required fields are available in the config.
config = _CONFIG.value
base_config.validate_config(config)

_CONFIG is None seems to be the problem. I'm using debugger from either PyCharm or VSCode.

Been having the same issue

the problem is ml_collections, the verison on pip is dated (even though it's 0.1.0 as required by jaxline)

install directly from source pip install git+https://github.com/google/ml_collections

someone from Google needs to push a new version of ml_collections to pip, and update the requirements here

The perceiver example from here works using absl's app:

from absl import app

# ...
# Experiment code
# ...

if __name__ == '__main__':
  flags.mark_flag_as_required('config')
  app.run(functools.partial(platform.main, Experiment))