ml

Contents

  • About
  • Getting Started

Configs

  • Launchers
  • LR Schedulers
  • Models
  • Optimizers

Reference

  • Core
    • ml.core.common_types
    • ml.core.config
    • ml.core.env
    • ml.core.registry
    • ml.core.state
  • Logger
  • LR Scheduler
  • Model
  • Optimizer
  • Task
  • Trainer
  • Launcher
  • Utility Functions
ml
  • Core
  • View page source

Core

Contents:

  • ml.core.common_types
  • ml.core.config
    • conf_field()
    • BaseConfig
    • BaseObject
  • ml.core.env
    • StrEnvVar
    • StrSetEnvVar
    • BoolEnvVar
    • IntEnvVar
    • PathEnvVar
    • is_debugging()
    • get_cache_dir()
    • get_run_dir()
    • set_run_dir()
    • get_eval_run_dir()
    • set_eval_run_dir()
    • get_exp_name()
    • set_exp_name()
    • get_data_dir()
    • set_data_dir()
    • get_slurm_conf_path()
    • get_s3_data_bucket()
    • set_s3_data_bucket()
    • get_s3_runs_bucket()
    • set_s3_runs_bucket()
    • get_model_dir()
    • set_model_dir()
    • get_env_random_seed()
    • set_env_random_seed()
    • get_stage_dir()
    • set_stage_dir()
    • get_global_tags()
    • set_global_tags()
    • add_global_tag()
    • is_metal_disabled()
    • is_gpu_disabled()
    • are_colors_disabled()
    • is_tensorboard_disabled()
    • should_show_full_import_error()
    • get_ml_config_path()
    • set_ml_config_path()
    • should_show_all_logs()
    • ignore_registry_cache()
    • get_wandb_entity()
    • get_default_config_root_path()
  • ml.core.registry
    • add_project_dir()
    • get_name()
    • get_names()
    • register_base
    • multi_register_base
    • register_model
    • register_task
    • register_trainer
    • register_optimizer
    • register_lr_scheduler
    • register_logger
    • register_launcher
    • Objects
  • ml.core.state
    • set_phase()
    • cast_phase()
    • State
Previous Next

© Copyright 2023, Benjamin Bolte.

Built with Sphinx using a theme provided by Read the Docs.