id: 1
unit: def set_up_data()
file: data.py
start line: 0
end line: 0
size: 74 LOC
McCabe index: 10
number of parameters: 1

id: 2
unit: def discretized_mix_logistic_loss()
file: vae_helpers.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 12
number of parameters: 3

id: 3
unit: def add_vae_arguments()
file: hps.py
start line: 0
end line: 0
size: 46 LOC
McCabe index: 1
number of parameters: 1

id: 4
unit: def logger()
file: utils.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 11
number of parameters: 1

id: 5
unit: def train_loop()
file: train.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 16
number of parameters: 7

id: 6
unit: def accumulate_stats()
file: train_helpers.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 13
number of parameters: 2

id: 7
unit: def load_vaes()
file: train_helpers.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 2

id: 8
unit: def cifar10()
file: data.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 2

id: 9
unit: def sample_from_discretized_mix_logistic()
file: vae_helpers.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 2

id: 10
unit: def training_step()
file: train.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 10
number of parameters: 7

id: 11
unit: def parse_layer_string()
file: vae.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 8
number of parameters: 1

id: 12
unit: def __init__()
file: vae.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 5

id: 13
unit: def maybe_download()
file: utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 2

id: 14
unit: def build()
file: vae.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 1

id: 15
unit: def tile_images()
file: utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 4

id: 16
unit: def set_up_hyperparams()
file: train_helpers.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 1

id: 17
unit: def write_images()
file: train.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 6

id: 18
unit: def build()
file: vae.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 1

id: 19
unit: def forward_uncond()
file: vae.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 4

id: 20
unit: def restore_params()
file: train_helpers.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 6

id: 21
unit: def load_opt()
file: train_helpers.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 3

id: 22
unit: def sample_uncond()
file: vae.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 4

id: 23
unit: def forward_uncond()
file: vae.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 4

id: 24
unit: def parse_args_and_update_hparams()
file: hps.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 3

id: 25
unit: def forward()
file: vae.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 2

id: 26
unit: def forward()
file: vae.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 4

id: 27
unit: def forward()
file: vae.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 3

id: 28
unit: def evaluate()
file: train.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 4

id: 29
unit: def setup_mpi()
file: train_helpers.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 1

id: 30
unit: def restore_log()
file: train_helpers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 12
number of parameters: 3

id: 31
unit: def forward()
file: vae.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 2

id: 32
unit: def get_inputs()
file: vae.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 3

id: 33
unit: def main()
file: train.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 0

id: 34
unit: def first_rank_first()
file: train_helpers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 2

id: 35
unit: def __init__()
file: vae.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 8

id: 36
unit: def get_width_settings()
file: vae.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 37
unit: def sample()
file: vae.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 38
unit: def forward()
file: vae.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 3

id: 39
unit: def forward_manual_latents()
file: vae.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 4

id: 40
unit: def imagenet32()
file: data.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 41
unit: def imagenet64()
file: data.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 42
unit: def get_conv()
file: vae_helpers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 9

id: 43
unit: def run_test_eval()
file: train.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 5

id: 44
unit: def save_model()
file: train_helpers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 45
unit: def distributed_maybe_download()
file: train_helpers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 46
unit: def ffhq256()
file: data.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 47
unit: def sample()
file: vae_helpers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 48
unit: def get_sample_for_visualization()
file: train.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 4

id: 49
unit: def unpickle_cifar10()
file: data.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 50
unit: def __init__()
file: vae_helpers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 51
unit: def __getattr__()
file: utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 52
unit: def num_nodes()
file: utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 53
unit: def gpus_per_node()
file: utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 54
unit: def eval_step()
file: train.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 55
unit: def __getattr__()
file: hps.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 56
unit: def pad_channels()
file: vae.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 57
unit: def log_prob_from_logits()
file: vae_helpers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 58
unit: def __init__()
file: vae_helpers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 59
unit: def allreduce()
file: utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 2

id: 60
unit: def get_cpu_stats_over_ranks()
file: utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 61
unit: def update_ema()
file: train_helpers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 62
unit: def linear_warmup()
file: train_helpers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 63
unit: def setup_save_dirs()
file: train_helpers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 64
unit: def forward_get_latents()
file: vae.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 65
unit: def draw_gaussian_diag_samples()
file: vae_helpers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 66
unit: def const_max()
file: vae_helpers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 67
unit: def const_min()
file: vae_helpers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 68
unit: def forward()
file: vae_helpers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 69
unit: def build()
file: vae.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 70
unit: def forward_uncond_samples()
file: vae.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 71
unit: def forward_samples_set_latents()
file: vae.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 72
unit: def gaussian_analytical_kl()
file: vae_helpers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 73
unit: def get_3x3()
file: vae_helpers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 6

id: 74
unit: def get_1x1()
file: vae_helpers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 6

id: 75
unit: def nll()
file: vae_helpers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 76
unit: def __setattr__()
file: utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 77
unit: def mpi_size()
file: utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 78
unit: def mpi_rank()
file: utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 79
unit: def local_mpi_rank()
file: utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 80
unit: def __setattr__()
file: hps.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 81
unit: def mkdir_p()
file: data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 82
unit: def flatten()
file: data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 83
unit: def ffhq1024()
file: data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1