Index

A

accel_hydra.models.common module accel_hydra.train_launcher module accel_hydra.utils.accelerate module accel_hydra.utils.config module

accel_hydra.utils.data module accel_hydra.utils.general module accel_hydra.utils.lr_scheduler module accel_hydra.utils.torch module AcceleratorSaveTrainableParams (class in accel_hydra.utils.accelerate)

C

concat_non_padding() (in module accel_hydra.utils.torch) contains_nan() (in module accel_hydra.utils.torch)

count_params() (accel_hydra.models.common.CountParamsMixin method) CountParamsMixin (class in accel_hydra.models.common) create_mask_from_length() (in module accel_hydra.utils.torch)

G

get_dataloader_one_pass_outside_steps() (in module accel_hydra.utils.lr_scheduler) get_dataloader_one_pass_steps_inside_accelerator() (in module accel_hydra.utils.lr_scheduler) get_dataloaders() (accel_hydra.train_launcher.TrainLauncher method) get_register_resolver_fn() (accel_hydra.train_launcher.TrainLauncher static method)

get_state_dict() (accel_hydra.utils.accelerate.AcceleratorSaveTrainableParams method) get_steps_for_lr_scheduler() (accel_hydra.train_launcher.TrainLauncher method) get_steps_inside_accelerator_from_outside_steps() (in module accel_hydra.utils.lr_scheduler) get_total_training_steps() (in module accel_hydra.utils.lr_scheduler) get_warmup_steps() (in module accel_hydra.utils.lr_scheduler)

I

init_dataloader_from_config() (in module accel_hydra.utils.data)

is_package_available() (in module accel_hydra.utils.general)

L

load_config_from_cli() (in module accel_hydra.utils.config) load_config_with_overrides() (in module accel_hydra.utils.config) load_pretrained() (accel_hydra.models.common.LoadPretrainedMixin method) load_pretrained_model() (in module accel_hydra.utils.torch)

load_state_dict() (accel_hydra.models.common.SaveTrainableParamsMixin method) LoadPretrainedMixin (class in accel_hydra.models.common) loss_with_mask() (in module accel_hydra.utils.torch) lr_scheduler_param_adapter() (in module accel_hydra.utils.lr_scheduler)

M

merge_matched_keys() (in module accel_hydra.utils.torch) module accel_hydra.models.common accel_hydra.train_launcher accel_hydra.utils.accelerate accel_hydra.utils.config accel_hydra.utils.data accel_hydra.utils.general accel_hydra.utils.lr_scheduler accel_hydra.utils.torch

multiply() (in module accel_hydra.utils.config)

P

param_names_to_save (accel_hydra.models.common.SaveTrainableParamsMixin property)

parse_launch_args() (in module accel_hydra.utils.config) process_state_dict() (accel_hydra.models.common.LoadPretrainedMixin method)

R

read_jsonl_to_mapping() (in module accel_hydra.utils.general) register_omegaconf_resolvers() (in module accel_hydra.utils.config)

remove_key_prefix_factory() (in module accel_hydra.utils.torch) restore_from_concat() (in module accel_hydra.utils.torch) run() (accel_hydra.train_launcher.TrainLauncher method)

S

SaveTrainableParamsMixin (class in accel_hydra.models.common)

setup_resume_cfg() (in module accel_hydra.utils.general)

T

TrainLauncher (class in accel_hydra.train_launcher)

trim_or_pad_length() (in module accel_hydra.utils.torch)