gflownet.utils.policy

Functions

parse_policy_config(config, kind)

Helper for parsing forward/backward policy_config from the given global config.

Module Contents

gflownet.utils.policy.parse_policy_config(config, kind)[source]

Helper for parsing forward/backward policy_config from the given global config.

Parameters:

  • config (DictConfig) – Global hydra config.

  • kind (str) – Type of config, either ‘forward’ or ‘backward’.

Return type:

Optional[omegaconf.DictConfig]