id: 1
unit: def get_cluster_input()
file: src/accelerate/commands/config/cluster.py
start line: 0
end line: 0
size: 772 LOC
McCabe index: 100
number of parameters: 0

id: 2
unit: def launch_command_parser()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 605 LOC
McCabe index: 6
number of parameters: 1

id: 3
unit: def _prepare_deepspeed()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 241 LOC
McCabe index: 86
number of parameters: 2

id: 4
unit: def prepare_model()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 193 LOC
McCabe index: 65
number of parameters: 4

id: 5
unit: def construct()
file: manim_animations/dataloaders/stage_5.py
start line: 0
end line: 0
size: 171 LOC
McCabe index: 20
number of parameters: 1

id: 6
unit: def _validate_launch_command()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 170 LOC
McCabe index: 86
number of parameters: 1

id: 7
unit: def get_sagemaker_input()
file: src/accelerate/commands/config/sagemaker.py
start line: 0
end line: 0
size: 164 LOC
McCabe index: 14
number of parameters: 0

id: 8
unit: def construct()
file: manim_animations/dataloaders/stage_6.py
start line: 0
end line: 0
size: 162 LOC
McCabe index: 20
number of parameters: 1

id: 9
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 160 LOC
McCabe index: 78
number of parameters: 1

id: 10
unit: def construct()
file: manim_animations/big_model_inference/stage_5.py
start line: 0
end line: 0
size: 159 LOC
McCabe index: 12
number of parameters: 1

id: 11
unit: def construct()
file: manim_animations/dataloaders/stage_7.py
start line: 0
end line: 0
size: 152 LOC
McCabe index: 15
number of parameters: 1

id: 12
unit: def prepare_deepspeed_cmd_env()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 140 LOC
McCabe index: 51
number of parameters: 1

id: 13
unit: def construct()
file: manim_animations/dataloaders/stage_2.py
start line: 0
end line: 0
size: 139 LOC
McCabe index: 11
number of parameters: 1

id: 14
unit: def prepare_multi_gpu_env()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 127 LOC
McCabe index: 44
number of parameters: 1

id: 15
unit: def __init__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 119 LOC
McCabe index: 46
number of parameters: 3

id: 16
unit: def construct()
file: manim_animations/big_model_inference/stage_3.py
start line: 0
end line: 0
size: 112 LOC
McCabe index: 14
number of parameters: 1

id: 17
unit: def construct()
file: manim_animations/big_model_inference/stage_4.py
start line: 0
end line: 0
size: 111 LOC
McCabe index: 14
number of parameters: 1

id: 18
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 101 LOC
McCabe index: 39
number of parameters: 1

id: 19
unit: def prepare()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 97 LOC
McCabe index: 46
number of parameters: 3

id: 20
unit: def get_max_memory()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 95 LOC
McCabe index: 46
number of parameters: 4

id: 21
unit: def load_state()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 94 LOC
McCabe index: 29
number of parameters: 4

id: 22
unit: def _prepare_megatron_lm()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 92 LOC
McCabe index: 35
number of parameters: 2

id: 23
unit: def prepare_simple_launcher_cmd_env()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 91 LOC
McCabe index: 34
number of parameters: 1

id: 24
unit: def construct()
file: manim_animations/big_model_inference/stage_2.py
start line: 0
end line: 0
size: 84 LOC
McCabe index: 11
number of parameters: 1

id: 25
unit: def write_basic_config()
file: src/accelerate/commands/config/default.py
start line: 0
end line: 0
size: 83 LOC
McCabe index: 17
number of parameters: 2

id: 26
unit: def save_state()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 81 LOC
McCabe index: 26
number of parameters: 4

id: 27
unit: def train_baseline()
file: benchmarks/fp8/torchao/distrib_deepspeed.py
start line: 0
end line: 0
size: 76 LOC
McCabe index: 6
number of parameters: 1

id: 28
unit: def load_state_dict()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 71 LOC
McCabe index: 30
number of parameters: 2

id: 29
unit: def synchronize_rng_state()
file: src/accelerate/utils/random.py
start line: 0
end line: 0
size: 69 LOC
McCabe index: 30
number of parameters: 2

id: 30
unit: def construct()
file: manim_animations/big_model_inference/stage_1.py
start line: 0
end line: 0
size: 69 LOC
McCabe index: 8
number of parameters: 1

id: 31
unit: def train_baseline()
file: benchmarks/fp8/transformer_engine/distrib_deepspeed.py
start line: 0
end line: 0
size: 68 LOC
McCabe index: 7
number of parameters: 1

id: 32
unit: def env_command()
file: src/accelerate/commands/env.py
start line: 0
end line: 0
size: 65 LOC
McCabe index: 23
number of parameters: 1

id: 33
unit: def skip_first_batches()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 61 LOC
McCabe index: 15
number of parameters: 2

id: 34
unit: def load_fsdp_model()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 59 LOC
McCabe index: 15
number of parameters: 6

id: 35
unit: def train_baseline()
file: benchmarks/fp8/ms_amp/distrib_deepspeed.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 4
number of parameters: 2

id: 36
unit: def fsdp2_prepare_model()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 27
number of parameters: 2

id: 37
unit: def tpu_command_parser()
file: src/accelerate/commands/tpu.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 3
number of parameters: 1

id: 38
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 19
number of parameters: 1

id: 39
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 33
number of parameters: 1

id: 40
unit: def main()
file: benchmarks/big_model_inference/big_model_inference.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 10
number of parameters: 0

id: 41
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 17
number of parameters: 1

id: 42
unit: def get_train_valid_test_datasets_provider()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 10
number of parameters: 2

id: 43
unit: def prepare_data_loader()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 8
number of parameters: 2

id: 44
unit: def convert_model()
file: src/accelerate/utils/transformer_engine.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 23
number of parameters: 4

id: 45
unit: def fsdp2_load_full_state_dict()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 13
number of parameters: 3

id: 46
unit: def tpu_pod_launcher()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 11
number of parameters: 1

id: 47
unit: def tpu_command_launcher()
file: src/accelerate/commands/tpu.py
start line: 0
end line: 0
size: 51 LOC
McCabe index: 24
number of parameters: 1

id: 48
unit: def create_empty_model()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 51 LOC
McCabe index: 14
number of parameters: 4

id: 49
unit: def model_provider_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 50 LOC
McCabe index: 8
number of parameters: 4

id: 50
unit: def _create_iam_role_for_sagemaker()
file: src/accelerate/commands/config/sagemaker.py
start line: 0
end line: 0
size: 49 LOC
McCabe index: 2
number of parameters: 1

id: 51
unit: def save_fsdp_model()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 11
number of parameters: 6

id: 52
unit: def train_integration()
file: benchmarks/fp8/torchao/distrib_deepspeed.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 2
number of parameters: 1

id: 53
unit: def get_batch_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 9
number of parameters: 3

id: 54
unit: def get_batch_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 6
number of parameters: 3

id: 55
unit: def deepspeed_launcher()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 14
number of parameters: 1

id: 56
unit: def _fetch_batches()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 18
number of parameters: 2

id: 57
unit: def main()
file: benchmarks/fsdp2/main.py
start line: 0
end line: 0
size: 46 LOC
McCabe index: 2
number of parameters: 0

id: 58
unit: def set_mixed_precision()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 46 LOC
McCabe index: 21
number of parameters: 4

id: 59
unit: def init_on_device()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 46 LOC
McCabe index: 12
number of parameters: 2

id: 60
unit: def split_between_processes()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 46 LOC
McCabe index: 19
number of parameters: 3

id: 61
unit: def load_fsdp_optimizer()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 45 LOC
McCabe index: 9
number of parameters: 7

id: 62
unit: def forward()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 45 LOC
McCabe index: 15
number of parameters: 2

id: 63
unit: def _prepare_msamp()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 18
number of parameters: 3

id: 64
unit: def train_baseline()
file: benchmarks/fp8/transformer_engine/fsdp.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 6
number of parameters: 0

id: 65
unit: def map_pytorch_optim_to_deepspeed()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 23
number of parameters: 1

id: 66
unit: def train_baseline()
file: benchmarks/fp8/torchao/fsdp.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 5
number of parameters: 0

id: 67
unit: def train_integration()
file: benchmarks/fp8/transformer_engine/distrib_deepspeed.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 3
number of parameters: 1

id: 68
unit: def get_batch_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 8
number of parameters: 3

id: 69
unit: def _prepare_ipex()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 25
number of parameters: 2

id: 70
unit: def get_state_dict()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 11
number of parameters: 3

id: 71
unit: def get_raw_chars()
file: src/accelerate/commands/menu/keymap.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 8
number of parameters: 0

id: 72
unit: def save_fsdp_optimizer()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 7
number of parameters: 6

id: 73
unit: def join_uneven_inputs()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 11
number of parameters: 3

id: 74
unit: def train_baseline()
file: benchmarks/fp8/transformer_engine/ddp.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 6
number of parameters: 0

id: 75
unit: def _prepare_fsdp2()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 28
number of parameters: 2

id: 76
unit: def init_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 19
number of parameters: 2

id: 77
unit: def prepare_dataloader()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 4
number of parameters: 3

id: 78
unit: def train_baseline()
file: benchmarks/fp8/torchao/ddp.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 5
number of parameters: 0

id: 79
unit: def pre_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 13
number of parameters: 4

id: 80
unit: def send_to_device()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 13
number of parameters: 4

id: 81
unit: def train_integration()
file: benchmarks/fp8/ms_amp/distrib_deepspeed.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 3
number of parameters: 2

id: 82
unit: def set_state_dict_type()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 15
number of parameters: 2

id: 83
unit: def get_loss_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 7
number of parameters: 4

id: 84
unit: def convert_config_to_fsdp2()
file: src/accelerate/commands/to_fsdp2.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 10
number of parameters: 1

id: 85
unit: def _iter_with_no_split()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 19
number of parameters: 1

id: 86
unit: def get_dataloaders()
file: benchmarks/fp8/torchao/fp8_utils.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 3
number of parameters: 2

id: 87
unit: def get_dataloaders()
file: benchmarks/fp8/transformer_engine/fp8_utils.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 3
number of parameters: 2

id: 88
unit: def train_baseline()
file: benchmarks/fp8/transformer_engine/non_distributed.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 5
number of parameters: 0

id: 89
unit: def train_integration()
file: benchmarks/fp8/transformer_engine/fsdp.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 3
number of parameters: 0

id: 90
unit: def get_dataloaders()
file: benchmarks/fp8/ms_amp/fp8_utils.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 3
number of parameters: 2

id: 91
unit: def get_mixed_precision_context_manager()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 8
number of parameters: 2

id: 92
unit: def _convert_nargs_to_dict()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 18
number of parameters: 1

id: 93
unit: def _prepare_te()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 15
number of parameters: 2

id: 94
unit: def clip_grad_norm_()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 19
number of parameters: 4

id: 95
unit: def parse_args()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 1
number of parameters: 0

id: 96
unit: def purge_accelerate_environment()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 13
number of parameters: 1

id: 97
unit: def pad_across_processes()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 12
number of parameters: 4

id: 98
unit: def create_ascii_table()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 9
number of parameters: 3

id: 99
unit: def train_baseline()
file: benchmarks/fp8/torchao/non_distributed.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 5
number of parameters: 0

id: 100
unit: def parse_bert_config()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 6
number of parameters: 3

id: 101
unit: def gather_for_metrics()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 10
number of parameters: 3

id: 102
unit: def construct()
file: manim_animations/dataloaders/stage_4.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 1
number of parameters: 1

id: 103
unit: def main()
file: utils/stale.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 14
number of parameters: 0

id: 104
unit: def train_integration()
file: benchmarks/fp8/torchao/fsdp.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 2
number of parameters: 0

id: 105
unit: def load_config_from_file()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 6
number of parameters: 1

id: 106
unit: def run()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 9
number of parameters: 2

id: 107
unit: def gather_data()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 7
number of parameters: 1

id: 108
unit: def multi_gpu_launcher()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 8
number of parameters: 1

id: 109
unit: def parse_args()
file: benchmarks/big_model_inference/big_model_inference.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 8
number of parameters: 0

id: 110
unit: def parse_gpt2_config()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 5
number of parameters: 3

id: 111
unit: def parse_llama_config()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 5
number of parameters: 3

id: 112
unit: def fsdp2_prepare_auto_wrap_policy()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 10
number of parameters: 2

id: 113
unit: def get_loss_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 7
number of parameters: 2

id: 114
unit: def initialize()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 13
number of parameters: 3

id: 115
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 8
number of parameters: 1

id: 116
unit: def compare_memory_usage()
file: benchmarks/fsdp2/visualize.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 3
number of parameters: 4

id: 117
unit: def train_integration()
file: benchmarks/fp8/transformer_engine/ddp.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 3
number of parameters: 0

id: 118
unit: def train_baseline()
file: benchmarks/fp8/ms_amp/ddp.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 2
number of parameters: 1

id: 119
unit: def parse_t5_config()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 6
number of parameters: 3

id: 120
unit: def recursively_apply()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 7
number of parameters: 6

id: 121
unit: def verify_operation()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 8
number of parameters: 1

id: 122
unit: def convert_file_size_to_int()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 13
number of parameters: 2

id: 123
unit: def prepare_model_optimizer_scheduler()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 5
number of parameters: 1

id: 124
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 6
number of parameters: 5

id: 125
unit: def eval_step()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 6
number of parameters: 2

id: 126
unit: def step()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 11
number of parameters: 2

id: 127
unit: def train_integration()
file: benchmarks/fp8/transformer_engine/non_distributed.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 2
number of parameters: 0

id: 128
unit: def train_integration()
file: benchmarks/fp8/ms_amp/ddp.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 2
number of parameters: 1

id: 129
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 11
number of parameters: 1

id: 130
unit: def set_scheduler_args()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 5
number of parameters: 2

id: 131
unit: def get_keys_to_not_convert()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 10
number of parameters: 1

id: 132
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 16
number of parameters: 1

id: 133
unit: def train_baseline()
file: benchmarks/fp8/ms_amp/non_distributed.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 2
number of parameters: 1

id: 134
unit: def train_integration()
file: benchmarks/fp8/ms_amp/non_distributed.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 2
number of parameters: 1

id: 135
unit: def register_comm_hook()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 5
number of parameters: 2

id: 136
unit: def set_mixed_precision()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 14
number of parameters: 2

id: 137
unit: def quantize_and_offload_8bit()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 9
number of parameters: 7

id: 138
unit: def ensure_weights_retied()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 14
number of parameters: 3

id: 139
unit: def merge_command_parser()
file: src/accelerate/commands/merge.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 3
number of parameters: 1

id: 140
unit: def post_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 12
number of parameters: 3

id: 141
unit: def train_integration()
file: benchmarks/fp8/torchao/ddp.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 2
number of parameters: 0

id: 142
unit: def pad_input_tensors()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 5
number of parameters: 4

id: 143
unit: def check_tied_parameters_in_config()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 10
number of parameters: 1

id: 144
unit: def add_hook_to_module()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 8
number of parameters: 3

id: 145
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 3
number of parameters: 1

id: 146
unit: def train_integration()
file: benchmarks/fp8/torchao/non_distributed.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 2
number of parameters: 0

id: 147
unit: def _deepspeed_config_checks()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 4
number of parameters: 1

id: 148
unit: def id_tensor_storage()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 5
number of parameters: 1

id: 149
unit: def get_grad_scaler()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 10
number of parameters: 2

id: 150
unit: def set_seed()
file: src/accelerate/utils/random.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 11
number of parameters: 3

id: 151
unit: def __call__()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 4
number of parameters: 3

id: 152
unit: def get_module_config()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 12
number of parameters: 1

id: 153
unit: def log_eval_results()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 8
number of parameters: 1

id: 154
unit: def default_command_parser()
file: src/accelerate/commands/config/default.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 1
number of parameters: 2

id: 155
unit: def estimate_command_parser()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 4
number of parameters: 1

id: 156
unit: def override_numa_affinity()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 9
number of parameters: 2

id: 157
unit: def _gpu_gather()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 10
number of parameters: 1

id: 158
unit: def get_state_dict_offloaded_model()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 9
number of parameters: 1

id: 159
unit: def align_module_device()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 8
number of parameters: 2

id: 160
unit: def get_batch_data_iterator()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 9
number of parameters: 2

id: 161
unit: def add_argument()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 14
number of parameters: 2

id: 162
unit: def launch_command()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 16
number of parameters: 1

id: 163
unit: def stop()
file: benchmarks/fsdp2/measure_utils.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 8
number of parameters: 1

id: 164
unit: def __init__()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 1
number of parameters: 0

id: 165
unit: def to_fsdp2_command_parser()
file: src/accelerate/commands/to_fsdp2.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 3
number of parameters: 1

id: 166
unit: def set_device()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 8
number of parameters: 1

id: 167
unit: def step()
file: src/accelerate/scheduler.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 12
number of parameters: 3

id: 168
unit: def pippy_forward()
file: src/accelerate/inference.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 5

id: 169
unit: def is_ipex_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 4
number of parameters: 0

id: 170
unit: def fill_match()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 9
number of parameters: 5

id: 171
unit: def set_deepspeed_weakref()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 9
number of parameters: 1

id: 172
unit: def set_auto_wrap_policy()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 8
number of parameters: 2

id: 173
unit: def decorator()
file: src/accelerate/utils/memory.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 2

id: 174
unit: def _prepare_ao()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 14
number of parameters: 2

id: 175
unit: def _get_named_parameters()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 8
number of parameters: 3

id: 176
unit: def process_config()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 14
number of parameters: 1

id: 177
unit: def tpu_launcher()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 4
number of parameters: 1

id: 178
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 9
number of parameters: 5

id: 179
unit: def get_module_children_bottom_up()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 9
number of parameters: 2

id: 180
unit: def __getitem__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 8
number of parameters: 2

id: 181
unit: def get_character()
file: src/accelerate/commands/menu/keymap.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 7
number of parameters: 0

id: 182
unit: def default_device()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 9
number of parameters: 1

id: 183
unit: def compile_regions()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 6
number of parameters: 2

id: 184
unit: def __init__()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 2

id: 185
unit: def retie_parameters()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 9
number of parameters: 2

id: 186
unit: def clear_device_cache()
file: src/accelerate/utils/memory.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 11
number of parameters: 1

id: 187
unit: def _add_action()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 4
number of parameters: 2

id: 188
unit: def __init__()
file: src/accelerate/local_sgd.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 4
number of parameters: 5

id: 189
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 2

id: 190
unit: def log()
file: src/accelerate/logging.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 5

id: 191
unit: def generate_device_map()
file: src/accelerate/inference.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 4

id: 192
unit: def clean_state_dict_for_safetensors()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 12
number of parameters: 1

id: 193
unit: def check_cuda_p2p_ib_support()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 8
number of parameters: 0

id: 194
unit: def set_moe_leaf_modules()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 2

id: 195
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 9
number of parameters: 1

id: 196
unit: def reduce()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 3

id: 197
unit: def apply_fp8_autowrap()
file: src/accelerate/utils/transformer_engine.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 2

id: 198
unit: def build_train_valid_test_data_iterators()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 2

id: 199
unit: def prepare_scheduler()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 2

id: 200
unit: def __init__()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 4

id: 201
unit: def config_command_parser()
file: src/accelerate/commands/config/config.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 1

id: 202
unit: def build()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 1

id: 203
unit: def deepspeed_config_process()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 6

id: 204
unit: def check_device_map()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 8
number of parameters: 5

id: 205
unit: def _distributed_checkpoint_to_merged_weights()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 3

id: 206
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 3

id: 207
unit: def offload_weight()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 4

id: 208
unit: def init_trackers()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 4

id: 209
unit: def _iter_with_split()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 10
number of parameters: 1

id: 210
unit: def construct()
file: manim_animations/dataloaders/stage_3.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 1

id: 211
unit: def parse_args()
file: benchmarks/fsdp2/visualize.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 212
unit: def get_training_utilities()
file: benchmarks/fp8/torchao/fp8_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 4

id: 213
unit: def get_training_utilities()
file: benchmarks/fp8/transformer_engine/fp8_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 3

id: 214
unit: def get_training_utilities()
file: benchmarks/fp8/ms_amp/fp8_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 3

id: 215
unit: def load()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 3

id: 216
unit: def patch_environment()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 1

id: 217
unit: def _get_profiler_activity()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 7
number of parameters: 2

id: 218
unit: def dtype_byte_size()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 8
number of parameters: 1

id: 219
unit: def fsdp2_apply_ac()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 220
unit: def load_checkpoint_and_dispatch()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 221
unit: def main()
file: src/accelerate/commands/accelerate_cli.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 0

id: 222
unit: def _convert_distributed_mode()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 1

id: 223
unit: def remove_hook_from_module()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 8
number of parameters: 2

id: 224
unit: def prepare_data_loader()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 225
unit: def notebook_launcher()
file: src/accelerate/launchers.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 2

id: 226
unit: def wait_for_everyone()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 1

id: 227
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 6

id: 228
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 1

id: 229
unit: def load_offloaded_weights()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 3

id: 230
unit: def install_xla()
file: src/accelerate/utils/torch_xla.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 231
unit: def get_xpu_available_memory()
file: src/accelerate/utils/memory.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 6
number of parameters: 1

id: 232
unit: def add_warning()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 2

id: 233
unit: def to_dict()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 8
number of parameters: 1

id: 234
unit: def move_direction()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 3

id: 235
unit: def estimate_training_usage()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 6
number of parameters: 3

id: 236
unit: def set_epoch()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 11
number of parameters: 2

id: 237
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 4

id: 238
unit: def get_cpu_distributed_information()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 239
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 1

id: 240
unit: def set_optimizer_type()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 2

id: 241
unit: def replace_with_bnb_layers()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 4

id: 242
unit: def calculate_maximum_sizes()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 1

id: 243
unit: def _handle_megatron_data_iterator()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 2

id: 244
unit: def load_offloaded_weight()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 245
unit: def trigger_sync_in_backward()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 1

id: 246
unit: def prepare_optimizer()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 7
number of parameters: 3

id: 247
unit: def backward()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 7
number of parameters: 3

id: 248
unit: def _get_devices()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 9
number of parameters: 2

id: 249
unit: def lomo_backward()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 6
number of parameters: 3

id: 250
unit: def fp8_backend()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 8
number of parameters: 1

id: 251
unit: def update_command_parser()
file: src/accelerate/commands/config/update.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 2

id: 252
unit: def __post_init__()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 7
number of parameters: 1

id: 253
unit: def select_row()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 1

id: 254
unit: def sagemaker_launcher()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 2

id: 255
unit: def log_images()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 5

id: 256
unit: def construct()
file: manim_animations/dataloaders/stage_0.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 1

id: 257
unit: def construct()
file: manim_animations/dataloaders/stage_1.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 1

id: 258
unit: def evaluate_model()
file: benchmarks/fp8/ms_amp/fp8_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 4

id: 259
unit: def find_pippy_batch_size()
file: src/accelerate/inference.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 8
number of parameters: 2

id: 260
unit: def build_pipeline()
file: src/accelerate/inference.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 5

id: 261
unit: def save()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 8
number of parameters: 4

id: 262
unit: def check_os_kernel()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 0

id: 263
unit: def is_bf16_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 1

id: 264
unit: def is_xpu_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 7
number of parameters: 1

id: 265
unit: def set_stage_and_offload()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 1

id: 266
unit: def del_config_sub_tree()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 7
number of parameters: 3

id: 267
unit: def set_cpu_offload()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 1

id: 268
unit: def validate_cpu_offload()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 1

id: 269
unit: def gather_tensor_shape()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 1

id: 270
unit: def find_device()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 8
number of parameters: 1

id: 271
unit: def GatheredParameters()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 4

id: 272
unit: def find_tied_parameters()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 13
number of parameters: 2

id: 273
unit: def clean_device_map()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 17
number of parameters: 5

id: 274
unit: def load_checkpoint_in_model()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 275
unit: def has_transformer_engine_layers()
file: src/accelerate/utils/transformer_engine.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 1

id: 276
unit: def contextual_fp8_autocast()
file: src/accelerate/utils/transformer_engine.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 3

id: 277
unit: def setup_fp8_env()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 3

id: 278
unit: def save_checkpoint()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 2

id: 279
unit: def on_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 4

id: 280
unit: def on_local_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 4

id: 281
unit: def accumulate()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 2

id: 282
unit: def _prepare_one()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 7
number of parameters: 4

id: 283
unit: def config_command()
file: src/accelerate/commands/config/config.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 1

id: 284
unit: def estimate_command()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 1

id: 285
unit: def adjust_state_dict_for_prefetch()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 7
number of parameters: 1

id: 286
unit: def debug_launcher()
file: src/accelerate/launchers.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 2

id: 287
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 288
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 2

id: 289
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 3

id: 290
unit: def filter_data()
file: benchmarks/fsdp2/visualize.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 4

id: 291
unit: def evaluate_model()
file: benchmarks/fp8/torchao/fp8_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 4

id: 292
unit: def evaluate_model()
file: benchmarks/fp8/torchao/non_distributed.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 4

id: 293
unit: def evaluate_model()
file: benchmarks/fp8/torchao/fsdp.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 4

id: 294
unit: def evaluate_model()
file: benchmarks/fp8/torchao/ddp.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 4

id: 295
unit: def evaluate_model()
file: benchmarks/fp8/transformer_engine/fp8_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 4

id: 296
unit: def is_musa_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 1

id: 297
unit: def is_npu_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 1

id: 298
unit: def is_sdaa_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 1

id: 299
unit: def deepspeed_required()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 1

id: 300
unit: def zero3_init_context_manager()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 2

id: 301
unit: def validate_mixed_precision_policy()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 1

id: 302
unit: def tqdm()
file: src/accelerate/utils/tqdm.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 3

id: 303
unit: def get_forward_step_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 3

id: 304
unit: def train_step()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 2

id: 305
unit: def load_checkpoint()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 2

id: 306
unit: def megatron_generate()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 307
unit: def save_offload_index()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 308
unit: def free_memory()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 2

id: 309
unit: def update_config()
file: src/accelerate/commands/config/update.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 1

id: 310
unit: def from_json_file()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 2

id: 311
unit: def from_yaml_file()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 2

id: 312
unit: def __len__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 1

id: 313
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 314
unit: def _check_initialized()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 3

id: 315
unit: def mixed_precision()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 1

id: 316
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 1

id: 317
unit: def _monitor()
file: benchmarks/fsdp2/measure_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 318
unit: def start()
file: benchmarks/fsdp2/measure_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 319
unit: def save_accelerator_state()
file: src/accelerate/checkpointing.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 320
unit: def load_accelerator_state()
file: src/accelerate/checkpointing.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 321
unit: def is_hpu_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 322
unit: def get_active_deepspeed_plugin()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 323
unit: def set_tensorboard_logging_options()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 324
unit: def get_module_class_from_name()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 2

id: 325
unit: def is_torch_xpu_tensor()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 1

id: 326
unit: def find_batch_size()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 7
number of parameters: 1

id: 327
unit: def chained_operation()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 328
unit: def _prepare_sd_options()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 329
unit: def gather_across_data_parallel_groups()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 330
unit: def extract_submodules_state_dict()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 8
number of parameters: 3

id: 331
unit: def on_main_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 332
unit: def on_local_main_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 333
unit: def on_last_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 334
unit: def register_for_checkpointing()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 2

id: 335
unit: def profile()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 2

id: 336
unit: def dispatch_model()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 337
unit: def zero_grad()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 2

id: 338
unit: def get_user_input()
file: src/accelerate/commands/config/config.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 0

id: 339
unit: def _ask_field()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 7
number of parameters: 4

id: 340
unit: def handle_input()
file: src/accelerate/commands/menu/input.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 341
unit: def env_command_parser()
file: src/accelerate/commands/env.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 342
unit: def check_has_model()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 7
number of parameters: 1

id: 343
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 344
unit: def __init__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 345
unit: def __init__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 2

id: 346
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 347
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 4

id: 348
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 4

id: 349
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 350
unit: def start_measure()
file: benchmarks/big_model_inference/measures_util.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 0

id: 351
unit: def end_measure()
file: benchmarks/big_model_inference/measures_util.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 1

id: 352
unit: def check_cuda_fp8_capability()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 0

id: 353
unit: def is_ccl_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 0

id: 354
unit: def is_megatron_lm_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 0

id: 355
unit: def is_mlflow_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 0

id: 356
unit: def set_network_size_args()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 3

id: 357
unit: def _tpu_gather()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 358
unit: def check_tied_parameters_on_same_device()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 2

id: 359
unit: def infer_auto_device_map()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 360
unit: def should_reduce_batch_size()
file: src/accelerate/utils/memory.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 1

id: 361
unit: def _get_mpirun_args()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 6
number of parameters: 0

id: 362
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 363
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 364
unit: def get_loss_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 365
unit: def get_forward_step_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 1

id: 366
unit: def unscale_gradients()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 7
number of parameters: 2

id: 367
unit: def __post_init__()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 1

id: 368
unit: def __new__()
file: src/accelerate/commands/menu/input.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 4

id: 369
unit: def hide_cursor()
file: src/accelerate/commands/menu/cursor.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 370
unit: def show_cursor()
file: src/accelerate/commands/menu/cursor.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 371
unit: def __init__()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 7

id: 372
unit: def __init__()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 373
unit: def attach_align_device_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 374
unit: def attach_align_device_hook_on_blocks()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 375
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 376
unit: def log_measures()
file: benchmarks/big_model_inference/measures_util.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 377
unit: def compile_regions_deepspeed()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 2

id: 378
unit: def convert_dict_to_env_variables()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 6
number of parameters: 1

id: 379
unit: def get_gpu_info()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 0

id: 380
unit: def is_torch_xla_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 2

id: 381
unit: def torchao_required()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 382
unit: def find_config_node()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 383
unit: def load_and_quantize_model()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 384
unit: def convert_to_fp32()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 385
unit: def get_non_persistent_buffers()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 3

id: 386
unit: def get_module_leaves()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 7
number of parameters: 1

id: 387
unit: def _get_model_state_dict()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 3

id: 388
unit: def _set_model_state_dict()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 4

id: 389
unit: def set_megatron_data_args()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 390
unit: def train()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 6
number of parameters: 1

id: 391
unit: def __init__()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 3

id: 392
unit: def simple_launcher()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 393
unit: def _set_sync_gradients()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 394
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 395
unit: def prepare_pippy()
file: src/accelerate/inference.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 5

id: 396
unit: def has_compiled_regions()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 1

id: 397
unit: def _recursive_unwrap()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 398
unit: def get_pretty_name()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 1

id: 399
unit: def merge_dicts()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 400
unit: def str_to_bool()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 2

id: 401
unit: def _nvidia_smi()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 0

id: 402
unit: def clear_environment()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 403
unit: def _is_package_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 404
unit: def add_model_config_to_megatron_parser()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 405
unit: def concatenate()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 8
number of parameters: 2

id: 406
unit: def check_device_same()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 8
number of parameters: 2

id: 407
unit: def set_module_tensor_to_device()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 408
unit: def _get_param_device()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 409
unit: def fsdp2_switch_optimizer_parameters()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 410
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 411
unit: def find_first_last_linear_layers()
file: src/accelerate/utils/ao.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 412
unit: def _do_sync()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 413
unit: def check_trigger()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 414
unit: def _attach_layerwise_casting_hooks()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 415
unit: def move_to_device()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 6
number of parameters: 2

id: 416
unit: def to_fsdp2_command()
file: src/accelerate/commands/to_fsdp2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 417
unit: def __init__()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 418
unit: def print_choice()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 419
unit: def verify_on_hub()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 420
unit: def __init__()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 3

id: 421
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 422
unit: def begin()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 423
unit: def set_sampler()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 424
unit: def __len__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 425
unit: def set_sampler()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 426
unit: def __repr__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 427
unit: def __getattr__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 428
unit: def __getattr__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 429
unit: def on_main_process()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 430
unit: def log_table()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 431
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 432
unit: def log_table()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 433
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 434
unit: def get_logger()
file: src/accelerate/logging.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 435
unit: def __init__()
file: benchmarks/fsdp2/measure_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 436
unit: def evaluate()
file: benchmarks/fsdp2/main.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 4

id: 437
unit: def filter_linear_layers()
file: benchmarks/fp8/torchao/non_distributed.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 7
number of parameters: 4

id: 438
unit: def filter_linear_layers()
file: benchmarks/fp8/torchao/fsdp.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 7
number of parameters: 4

id: 439
unit: def filter_linear_layers()
file: benchmarks/fp8/torchao/distrib_deepspeed.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 7
number of parameters: 4

id: 440
unit: def filter_linear_layers()
file: benchmarks/fp8/torchao/ddp.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 7
number of parameters: 4

id: 441
unit: def convert_bytes()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 442
unit: def is_bnb_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 443
unit: def is_mlu_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 444
unit: def select()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 445
unit: def set_mixed_precision()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 446
unit: def set_training_args()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 447
unit: def listify()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 448
unit: def gather()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 449
unit: def gather_object()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 450
unit: def broadcast()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 451
unit: def broadcast_object_list()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 2

id: 452
unit: def get_balanced_memory()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 453
unit: def fsdp2_canonicalize_names()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 1

id: 454
unit: def release_memory()
file: src/accelerate/utils/memory.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 455
unit: def compare_versions()
file: src/accelerate/utils/versions.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 4

id: 456
unit: def get_enc_dec_mask()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 457
unit: def filter_linear_layers()
file: src/accelerate/utils/ao.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 3

id: 458
unit: def no_sync()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 2

id: 459
unit: def get_tracker()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 3

id: 460
unit: def save()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 4

id: 461
unit: def autocast()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 462
unit: def attach_layerwise_casting_hooks()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 463
unit: def __getstate__()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 464
unit: def _validate_to_fsdp2_args()
file: src/accelerate/commands/to_fsdp2.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 6
number of parameters: 1

id: 465
unit: def get_config_parser()
file: src/accelerate/commands/config/__init__.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 466
unit: def main()
file: src/accelerate/commands/config/__init__.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 467
unit: def mark()
file: src/accelerate/commands/menu/input.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 468
unit: def mark_multiple()
file: src/accelerate/commands/menu/input.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 469
unit: def hide()
file: src/accelerate/commands/menu/cursor.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 470
unit: def __init__()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 471
unit: def add_argument()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 3

id: 472
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 473
unit: def total_batch_size()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 474
unit: def set_epoch()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 2

id: 475
unit: def get_sampler()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 476
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 477
unit: def is_local_main_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 478
unit: def __repr__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 479
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 480
unit: def peak_monitor()
file: benchmarks/big_model_inference/measures_util.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

id: 481
unit: def __init__()
file: src/accelerate/scheduler.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 5

id: 482
unit: def has_repeated_blocks()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

id: 483
unit: def model_has_dtensor()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 484
unit: def get_int_from_env()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 485
unit: def get_driver_version()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 486
unit: def is_xccl_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 0

id: 487
unit: def is_torchao_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 488
unit: def is_4bit_bnb_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 489
unit: def is_8bit_bnb_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 490
unit: def is_aim_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 491
unit: def is_habana_gaudi1()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 0

id: 492
unit: def is_torchdata_stateful_dataloader_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 493
unit: def __init__()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 6

id: 494
unit: def to_kwargs()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 495
unit: def __contains__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 496
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 497
unit: def is_auto()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 498
unit: def has_4bit_bnb_layers()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 499
unit: def ignorant_find_batch_size()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 500
unit: def _tpu_broadcast()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 3

id: 501
unit: def copy_tensor_to_devices()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 502
unit: def convert_outputs_to_fp32()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 503
unit: def values()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 504
unit: def _init_infer_auto_device_map()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 505
unit: def fallback_allocate()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 506
unit: def _filter_args()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 507
unit: def get_forward_step_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 508
unit: def attn_mask_postprocess()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 509
unit: def finish_mpu_init()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 0

id: 510
unit: def train()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 511
unit: def offload_state_dict()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 512
unit: def __init__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 513
unit: def has_ao_layers()
file: src/accelerate/utils/ao.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 514
unit: def _prepare_device_mesh()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

id: 515
unit: def save_model()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 516
unit: def cpu_offload()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 517
unit: def disk_offload()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 518
unit: def __setstate__()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 519
unit: def load_config()
file: src/accelerate/commands/to_fsdp2.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 520
unit: def clean_option()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 521
unit: def __repr__()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 522
unit: def detach_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 523
unit: def attach_execution_device_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 524
unit: def pre_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 525
unit: def step()
file: src/accelerate/local_sgd.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 526
unit: def _goes_first()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 527
unit: def on_main_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 3

id: 528
unit: def on_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 4

id: 529
unit: def on_local_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 4

id: 530
unit: def train()
file: benchmarks/fsdp2/main.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 531
unit: def get_named_parameters()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 2

id: 532
unit: def replace_optimizer_params()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 6
number of parameters: 1

id: 533
unit: def start()
file: benchmarks/big_model_inference/measures_util.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 534
unit: def is_port_in_use()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 535
unit: def is_transformer_engine_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 536
unit: def is_fp16_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 537
unit: def is_bitsandbytes_multi_backend_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 538
unit: def get_value()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 539
unit: def backward()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 4

id: 540
unit: def get_global_grad_norm()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 541
unit: def __init__()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 5

id: 542
unit: def set_directories()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 543
unit: def _replace_with_bnb_layers()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 544
unit: def honor_type()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 545
unit: def _gpu_broadcast()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 546
unit: def is_peft_model()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 547
unit: def _get_proper_dtype()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 548
unit: def compute_module_sizes()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 549
unit: def get_module_size_with_ties()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 550
unit: def get_state_dict_from_offload()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 551
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 5

id: 552
unit: def eval()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 553
unit: def clip_grad_value_()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 554
unit: def optimizer_step_was_skipped()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 555
unit: def verify_device_map()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 6
number of parameters: 2

id: 556
unit: def init_empty_weights()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 557
unit: def patch_optimizer_step()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 558
unit: def write_choice()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 559
unit: def main()
file: src/accelerate/commands/env.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 560
unit: def __call__()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 5

id: 561
unit: def end_section()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 562
unit: def __enter__()
file: src/accelerate/local_sgd.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 563
unit: def _sync_and_avg_model_params()
file: src/accelerate/local_sgd.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 564
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 565
unit: def __len__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 566
unit: def _update_state_dict()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 1

id: 567
unit: def total_dataset_length()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 568
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 569
unit: def is_main_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 570
unit: def destroy_process_group()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 2

id: 571
unit: def __repr__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 572
unit: def _reset_state()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 573
unit: def deepspeed_plugin()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 574
unit: def select_deepspeed_plugin()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 575
unit: def end_of_dataloader()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 576
unit: def remainder()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 577
unit: def is_xla_gradients_synced()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 578
unit: def _remove_dataloader()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 579
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 5

id: 580
unit: def log_images()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 4

id: 581
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 5

id: 582
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 583
unit: def log_images()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 4

id: 584
unit: def _get_title_series()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 585
unit: def log_images()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 4

id: 586
unit: def get_model()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 587
unit: def get_tokenizer()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 588
unit: def stop()
file: benchmarks/big_model_inference/measures_util.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 589
unit: def get_named_parameters()
file: benchmarks/fp8/torchao/fp8_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 590
unit: def get_named_parameters()
file: benchmarks/fp8/transformer_engine/fp8_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 591
unit: def get_named_parameters()
file: benchmarks/fp8/ms_amp/fp8_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 592
unit: def save_custom_state()
file: src/accelerate/checkpointing.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 593
unit: def load_custom_state()
file: src/accelerate/checkpointing.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 594
unit: def is_compiled_module()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 595
unit: def get_free_port()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 596
unit: def recursive_getattr()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 597
unit: def is_cuda_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 598
unit: def is_triton_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 599
unit: def is_rich_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 600
unit: def step_was_skipped()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 601
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 602
unit: def to_dict()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 603
unit: def to_kwargs()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 604
unit: def selected()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 605
unit: def get_data_structure()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 606
unit: def get_shape()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 607
unit: def initialize_tensors()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 608
unit: def _gpu_gather_object()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 4
number of parameters: 1

id: 609
unit: def slice_tensors()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 610
unit: def __getstate__()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 611
unit: def compute_module_total_buffer_size()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 612
unit: def enable_fsdp_ram_efficient_loading()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 0

id: 613
unit: def reduce_batch_size_fn()
file: src/accelerate/utils/memory.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 614
unit: def env_var_path_add()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 2

id: 615
unit: def __init__()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 616
unit: def prepare_optimizer()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 617
unit: def prepare_scheduler()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 618
unit: def get_decoder_mask()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 619
unit: def convert_model_to_fp8_ao()
file: src/accelerate/utils/ao.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 620
unit: def use_stateful_dataloader()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 621
unit: def end_training()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 622
unit: def register_save_state_pre_hook()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 623
unit: def register_load_state_pre_hook()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 624
unit: def cpu_offload_with_hook()
file: src/accelerate/big_modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 625
unit: def load_state_dict()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 2

id: 626
unit: def default_config_command()
file: src/accelerate/commands/config/default.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 627
unit: def main()
file: src/accelerate/commands/config/config.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 628
unit: def to_json_file()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 629
unit: def _ask_options()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 630
unit: def _format_usage()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 5

id: 631
unit: def main()
file: src/accelerate/commands/tpu.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 632
unit: def main()
file: src/accelerate/commands/estimate.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 633
unit: def merge_command()
file: src/accelerate/commands/merge.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 634
unit: def main()
file: src/accelerate/commands/merge.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 635
unit: def add_argument_group()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 636
unit: def main()
file: src/accelerate/commands/launch.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 637
unit: def init_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 638
unit: def pre_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 639
unit: def post_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 640
unit: def detach_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 641
unit: def remove_hook_from_submodules()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 642
unit: def __init__()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 643
unit: def __init__()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 644
unit: def __exit__()
file: src/accelerate/local_sgd.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 645
unit: def set_epoch()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 646
unit: def __getattr__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 2

id: 647
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 648
unit: def total_batch_size()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 649
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 650
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 5

id: 651
unit: def on_local_main_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 652
unit: def on_last_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 653
unit: def dataloader_references()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 2

id: 654
unit: def log_images()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 655
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 656
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 657
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 658
unit: def log_figure()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 659
unit: def log_artifacts()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 660
unit: def log_artifact()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 661
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 662
unit: def __init__()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 663
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 664
unit: def _should_log()
file: src/accelerate/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 665
unit: def __init__()
file: benchmarks/big_model_inference/measures_util.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 666
unit: def parse_flag_from_env()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 667
unit: def parse_choice_from_env()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 668
unit: def is_fp8_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 0

id: 669
unit: def is_mps_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 670
unit: def is_numpy_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 671
unit: def is_true()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 672
unit: def is_false()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 673
unit: def __init__()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 674
unit: def list()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 675
unit: def __init__()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 676
unit: def has_offloaded_params()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 677
unit: def get_fsdp2_grad_scaler()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 678
unit: def synchronize_rng_states()
file: src/accelerate/utils/random.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 679
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 680
unit: def __init__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 681
unit: def filter_first_and_last_linear_layers()
file: src/accelerate/utils/ao.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 682
unit: def split_between_processes()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 683
unit: def main_process_first()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 684
unit: def local_main_process_first()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 685
unit: def log()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 4

id: 686
unit: def __deepcopy__()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 687
unit: def eval()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 688
unit: def _switch_parameters()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 2

id: 689
unit: def update_config_command()
file: src/accelerate/commands/config/update.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 690
unit: def _get_iam_role_arn()
file: src/accelerate/commands/config/sagemaker.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 691
unit: def to_yaml_file()
file: src/accelerate/commands/config/config_args.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 692
unit: def _convert_compute_environment()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 693
unit: def _convert_dynamo_backend()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 694
unit: def _convert_mixed_precision()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 695
unit: def _convert_sagemaker_distributed_mode()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 696
unit: def _convert_fp8_backend()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 697
unit: def forceWrite()
file: src/accelerate/commands/menu/helpers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 698
unit: def clear_line()
file: src/accelerate/commands/menu/helpers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 699
unit: def linebreak()
file: src/accelerate/commands/menu/helpers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 700
unit: def select()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 701
unit: def interrupt()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 702
unit: def __init__()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 703
unit: def init_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 704
unit: def pre_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 705
unit: def post_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 706
unit: def set_epoch()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 707
unit: def __init_subclass__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 708
unit: def reset()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 709
unit: def end()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 710
unit: def __reduce__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 711
unit: def set_epoch()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 712
unit: def __reduce__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 713
unit: def __init__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 714
unit: def __reduce__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 715
unit: def test_launch()
file: src/accelerate/launchers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 716
unit: def _reset_state()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 717
unit: def initialized()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 718
unit: def is_last_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 719
unit: def print()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 720
unit: def is_last_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 721
unit: def is_main_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 722
unit: def is_local_main_process()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 723
unit: def split_between_processes()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 724
unit: def main_process_first()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 725
unit: def local_main_process_first()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 726
unit: def num_steps()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 727
unit: def adjust_scheduler()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 728
unit: def sync_with_dataloader()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 729
unit: def initialized()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 730
unit: def is_xla_gradients_synced()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 731
unit: def _add_dataloader()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 732
unit: def in_dataloader()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 733
unit: def _reset_state()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 734
unit: def get_available_trackers()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 735
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 736
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 737
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 738
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 739
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 740
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 4

id: 741
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 742
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 743
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 744
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 745
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 746
unit: def filter_trackers()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 747
unit: def peak_allocated_memory()
file: benchmarks/fsdp2/measure_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 748
unit: def peak_reserved_memory()
file: benchmarks/fsdp2/measure_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 749
unit: def swap_back_optimizer_params()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 750
unit: def prepare_torch()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 751
unit: def prepare_accelerate()
file: benchmarks/fsdp2/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 752
unit: def get_last_lr()
file: src/accelerate/scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 753
unit: def state_dict()
file: src/accelerate/scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 754
unit: def load_state_dict()
file: src/accelerate/scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 755
unit: def get_lr()
file: src/accelerate/scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 756
unit: def print_lr()
file: src/accelerate/scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 757
unit: def forward()
file: src/accelerate/inference.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 758
unit: def is_repeated_blocks()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 759
unit: def extract_model_from_parallel()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 760
unit: def wait_for_everyone()
file: src/accelerate/utils/other.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 761
unit: def are_libraries_initialized()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 762
unit: def set_numa_affinity()
file: src/accelerate/utils/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 763
unit: def is_torch_distributed_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 764
unit: def get_ccl_version()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 765
unit: def is_import_timer_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 766
unit: def is_pynvml_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 0

id: 767
unit: def is_pytest_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 768
unit: def is_msamp_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 769
unit: def is_schedulefree_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 770
unit: def is_lomo_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 771
unit: def is_deepspeed_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 772
unit: def is_pippy_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 773
unit: def is_torchvision_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 774
unit: def is_transformers_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 775
unit: def is_datasets_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 776
unit: def is_peft_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 777
unit: def is_timm_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 778
unit: def is_tensorboard_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 0

id: 779
unit: def is_wandb_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 780
unit: def is_comet_ml_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 781
unit: def is_swanlab_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 782
unit: def is_boto3_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 783
unit: def is_sagemaker_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 784
unit: def is_tqdm_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 785
unit: def is_clearml_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 786
unit: def is_pandas_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 787
unit: def is_matplotlib_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 788
unit: def is_dvclive_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 789
unit: def is_torchdata_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 790
unit: def is_weights_only_available()
file: src/accelerate/utils/imports.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 791
unit: def is_zero2()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 792
unit: def is_zero3()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 793
unit: def is_offload()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 794
unit: def __init__()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 795
unit: def zero_grad()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 796
unit: def step()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 797
unit: def __init__()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 798
unit: def step()
file: src/accelerate/utils/deepspeed.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 799
unit: def to_dict()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 800
unit: def __str__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 801
unit: def to_dict()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 4

id: 802
unit: def __post_init__()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 803
unit: def get_value()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 804
unit: def is_zero3_init_enabled()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 805
unit: def _unselect()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 806
unit: def _set_selected()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 807
unit: def selected()
file: src/accelerate/utils/dataclasses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 808
unit: def get_quantized_model_device_map()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 809
unit: def get_parameter_device()
file: src/accelerate/utils/bnb.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 810
unit: def is_torch_tensor()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 811
unit: def is_tensor_information()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 812
unit: def is_namedtuple()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 813
unit: def __call__()
file: src/accelerate/utils/operations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 814
unit: def named_module_tensors()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 815
unit: def __init__()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 816
unit: def get_max_layer_size()
file: src/accelerate/utils/modeling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 817
unit: def disable_fsdp_ram_efficient_loading()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 818
unit: def merge_fsdp_weights()
file: src/accelerate/utils/fsdp_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 819
unit: def find_executable_batch_size()
file: src/accelerate/utils/memory.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 820
unit: def is_torch_version()
file: src/accelerate/utils/versions.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 821
unit: def prepare_tpu()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 822
unit: def prepare_sagemager_args_inputs()
file: src/accelerate/utils/launch.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 823
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 824
unit: def zero_grad()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 825
unit: def step()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 826
unit: def step_was_skipped()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 827
unit: def __init__()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 828
unit: def step()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 829
unit: def get_batch_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 830
unit: def get_forward_step_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 831
unit: def get_loss_func()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 832
unit: def avg_losses_across_data_parallel_group()
file: src/accelerate/utils/megatron_lm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 833
unit: def __getitem__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 834
unit: def __iter__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 835
unit: def __len__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 836
unit: def __iter__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 837
unit: def __len__()
file: src/accelerate/utils/offload.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 838
unit: def deepspeed_plugin()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 839
unit: def use_distributed()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 840
unit: def distributed_type()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 841
unit: def num_processes()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 842
unit: def process_index()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 843
unit: def local_process_index()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 844
unit: def device()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 845
unit: def split_batches()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 846
unit: def dispatch_batches()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 847
unit: def even_batches()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 848
unit: def even_batches()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 849
unit: def use_seedable_sampler()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 850
unit: def non_blocking()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 851
unit: def project_dir()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 852
unit: def logging_dir()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 853
unit: def save_iteration()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 854
unit: def is_main_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 855
unit: def is_local_main_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 856
unit: def is_last_process()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 857
unit: def mixed_precision()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 858
unit: def is_fsdp2()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 859
unit: def sync_gradients()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 860
unit: def sync_gradients()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 861
unit: def gradient_accumulation_steps()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 862
unit: def gradient_accumulation_steps()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 863
unit: def print()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 864
unit: def prepare_data_loader()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 865
unit: def set_trigger()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 866
unit: def gather()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 867
unit: def reduce()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 868
unit: def pad_across_processes()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 5

id: 869
unit: def unwrap_model()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 870
unit: def wait_for_everyone()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 871
unit: def clear()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 872
unit: def skip_first_batches()
file: src/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 873
unit: def state()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 874
unit: def state()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 875
unit: def param_groups()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 876
unit: def param_groups()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 877
unit: def defaults()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 878
unit: def defaults()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 879
unit: def add_param_group()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 880
unit: def state_dict()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 881
unit: def step_was_skipped()
file: src/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 882
unit: def _convert_yes_no_to_bool()
file: src/accelerate/commands/config/config_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 883
unit: def register()
file: src/accelerate/commands/menu/input.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 884
unit: def writeColor()
file: src/accelerate/commands/menu/helpers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 885
unit: def reset_cursor()
file: src/accelerate/commands/menu/helpers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 886
unit: def move_cursor()
file: src/accelerate/commands/menu/helpers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 887
unit: def move_up()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 888
unit: def move_down()
file: src/accelerate/commands/menu/selection_menu.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 889
unit: def __call__()
file: src/accelerate/commands/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 5

id: 890
unit: def init_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 891
unit: def pre_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 892
unit: def post_forward()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 893
unit: def detach_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 894
unit: def __init__()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 895
unit: def init_hook()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 896
unit: def offload()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 897
unit: def remove()
file: src/accelerate/hooks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 898
unit: def total_length()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 899
unit: def __iter__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 900
unit: def state_dict()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 901
unit: def load_state_dict()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 902
unit: def __class__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 903
unit: def __len__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 904
unit: def get_sampler()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 905
unit: def total_batch_size()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 906
unit: def total_dataset_length()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 907
unit: def batch_sampler()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 908
unit: def dataloader()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 909
unit: def total_dataset_length()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 910
unit: def get_sampler()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 911
unit: def total_length()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 912
unit: def __len__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 913
unit: def __len__()
file: src/accelerate/data_loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 914
unit: def is_initialized()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 915
unit: def do_nothing()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 916
unit: def __init__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 917
unit: def __get__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 918
unit: def __set__()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 919
unit: def use_distributed()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 920
unit: def main_process_first()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 921
unit: def local_main_process_first()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 922
unit: def _prepare_backend()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 923
unit: def initialized()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 924
unit: def destroy_process_group()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 925
unit: def fork_launched()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 926
unit: def use_distributed()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 927
unit: def is_fsdp2()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 928
unit: def wait_for_everyone()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 929
unit: def get_deepspeed_plugin()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 930
unit: def print()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 931
unit: def active_dataloader()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 932
unit: def dataloader_references()
file: src/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 4
number of parameters: 1

id: 933
unit: def start()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 934
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 935
unit: def log()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 936
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 937
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 938
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 939
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 940
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 941
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 942
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 943
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 944
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 945
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 946
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 947
unit: def store_init_configuration()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 948
unit: def finish()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 949
unit: def tracker()
file: src/accelerate/tracking.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 950
unit: def warning_once()
file: src/accelerate/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3