()
| 24 | |
| 25 | |
| 26 | def get_config(): |
| 27 | config_path = os.environ.get("OMINI_CONFIG") |
| 28 | assert config_path is not None, "Please set the OMINI_CONFIG environment variable" |
| 29 | with open(config_path, "r") as f: |
| 30 | config = yaml.safe_load(f) |
| 31 | return config |
| 32 | |
| 33 | |
| 34 | def init_wandb(wandb_config, run_name): |