id: 1
unit: def evaluate()
file: jcm/evaluate.py
start line: 0
end line: 0
size: 292 LOC
McCabe index: 50
number of parameters: 3

id: 2
unit: def train()
file: jcm/train.py
start line: 0
end line: 0
size: 236 LOC
McCabe index: 47
number of parameters: 2

id: 3
unit: def __call__()
file: jcm/models/ncsnpp.py
start line: 0
end line: 0
size: 180 LOC
McCabe index: 36
number of parameters: 4

id: 4
unit: def get_loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 136 LOC
McCabe index: 8
number of parameters: 5

id: 5
unit: def loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 127 LOC
McCabe index: 10
number of parameters: 4

id: 6
unit: def step_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 83 LOC
McCabe index: 6
number of parameters: 2

id: 7
unit: def dsm_loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 80 LOC
McCabe index: 4
number of parameters: 4

id: 8
unit: def ssm_loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 77 LOC
McCabe index: 4
number of parameters: 4

id: 9
unit: def loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 73 LOC
McCabe index: 9
number of parameters: 6

id: 10
unit: def __call__()
file: jcm/models/ncsnv2.py
start line: 0
end line: 0
size: 69 LOC
McCabe index: 2
number of parameters: 4

id: 11
unit: def get_config()
file: configs/cifar10_ve_ct_adaptive.py
start line: 0
end line: 0
size: 67 LOC
McCabe index: 2
number of parameters: 0

id: 12
unit: def __call__()
file: jcm/models/ncsnv2.py
start line: 0
end line: 0
size: 65 LOC
McCabe index: 2
number of parameters: 4

id: 13
unit: def get_config()
file: configs/cifar10_ve_ct_ema.py
start line: 0
end line: 0
size: 65 LOC
McCabe index: 2
number of parameters: 0

id: 14
unit: def get_config()
file: configs/cifar10_ve_progressive_distillation.py
start line: 0
end line: 0
size: 65 LOC
McCabe index: 1
number of parameters: 0

id: 15
unit: def get_config()
file: configs/cifar10_ve_cd.py
start line: 0
end line: 0
size: 63 LOC
McCabe index: 2
number of parameters: 0

id: 16
unit: def loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 61 LOC
McCabe index: 15
number of parameters: 6

id: 17
unit: def get_config()
file: configs/cifar10_ve_continuous.py
start line: 0
end line: 0
size: 60 LOC
McCabe index: 2
number of parameters: 0

id: 18
unit: def get_sampling_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 58 LOC
McCabe index: 9
number of parameters: 5

id: 19
unit: def get_optimizer()
file: jcm/losses.py
start line: 0
end line: 0
size: 58 LOC
McCabe index: 6
number of parameters: 1

id: 20
unit: def __call__()
file: jcm/models/ncsnv2.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 2
number of parameters: 4

id: 21
unit: def get_config()
file: configs/cifar10_k_ve.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 2
number of parameters: 0

id: 22
unit: def get_config()
file: configs/cifar10_k_vp.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 2
number of parameters: 0

id: 23
unit: def get_default_configs()
file: configs/default_cifar10_configs.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 2
number of parameters: 0

id: 24
unit: def get_ema_scales_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 54 LOC
McCabe index: 8
number of parameters: 1

id: 25
unit: def upfirdn_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 8
number of parameters: 10

id: 26
unit: def __call__()
file: jcm/models/ddpm.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 12
number of parameters: 4

id: 27
unit: def __call__()
file: jcm/models/ncsnv2.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 2
number of parameters: 4

id: 28
unit: def collate_fn()
file: jcm/datasets.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 4
number of parameters: 1

id: 29
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 4
number of parameters: 3

id: 30
unit: def __call__()
file: jcm/models/normalization.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 3
number of parameters: 3

id: 31
unit: def get_heun_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 2
number of parameters: 4

id: 32
unit: def get_score_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 10
number of parameters: 6

id: 33
unit: def likelihood_fn()
file: jcm/likelihood.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 4
number of parameters: 3

id: 34
unit: def get_euler_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 2
number of parameters: 4

id: 35
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 7
number of parameters: 2

id: 36
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 7
number of parameters: 3

id: 37
unit: def save_image()
file: jcm/utils.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 10
number of parameters: 6

id: 38
unit: def __call__()
file: jcm/models/layerspp.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 11
number of parameters: 4

id: 39
unit: def __call__()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 1
number of parameters: 4

id: 40
unit: def get_progressive_distillation_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 2
number of parameters: 5

id: 41
unit: def main()
file: jcm/main.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 4
number of parameters: 1

id: 42
unit: def upsample_conv_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 4
number of parameters: 6

id: 43
unit: def __call__()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 4
number of parameters: 4

id: 44
unit: def get_sde()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 5
number of parameters: 1

id: 45
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 3
number of parameters: 4

id: 46
unit: def reverse()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 7
number of parameters: 3

id: 47
unit: def ode_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 4
number of parameters: 3

id: 48
unit: def denoiser_distiller_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 4
number of parameters: 3

id: 49
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 4
number of parameters: 2

id: 50
unit: def __call__()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 6
number of parameters: 2

id: 51
unit: def __call__()
file: jcm/models/layerspp.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 2

id: 52
unit: def marginal_prob()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 2
number of parameters: 4

id: 53
unit: def marginal_prob()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 2
number of parameters: 4

id: 54
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 3
number of parameters: 4

id: 55
unit: def __call__()
file: jcm/models/layerspp.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 5
number of parameters: 2

id: 56
unit: def __call__()
file: jcm/models/layerspp.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 6
number of parameters: 4

id: 57
unit: def _simple_upfirdn_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 3
number of parameters: 7

id: 58
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 2

id: 59
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 2

id: 60
unit: def get_normalization()
file: jcm/models/normalization.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 7
number of parameters: 2

id: 61
unit: def get_fids()
file: jcm/metrics.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 4

id: 62
unit: def init_model()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 2

id: 63
unit: def get_denoiser_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 6

id: 64
unit: def conv_downsample_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 3
number of parameters: 6

id: 65
unit: def shake_drop_train()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 7

id: 66
unit: def ncsn_conv1x1()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 6

id: 67
unit: def ncsn_conv3x3()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 6

id: 68
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 4

id: 69
unit: def save_checkpoint()
file: jcm/checkpoints.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 5

id: 70
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 3

id: 71
unit: def __call__()
file: jcm/models/normalization.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 2

id: 72
unit: def __call__()
file: jcm/models/layerspp.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 2

id: 73
unit: def distiller_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 3

id: 74
unit: def get_logit_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 2

id: 75
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 2

id: 76
unit: def upsample_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 5

id: 77
unit: def __call__()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 3

id: 78
unit: def likelihood_importance_cum_weight()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 3

id: 79
unit: def likelihood_importance_cum_weight()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 3

id: 80
unit: def vesde_update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 4

id: 81
unit: def pc_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 2

id: 82
unit: def get_onestep_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 4

id: 83
unit: def get_seeded_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 4

id: 84
unit: def heun_solver()
file: jcm/losses.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 4

id: 85
unit: def register_model()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 86
unit: def create_classifier()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 3

id: 87
unit: def shake_shake_train()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 6
number of parameters: 3

id: 88
unit: def discretize()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 3

id: 89
unit: def register_predictor()
file: jcm/sampling.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 90
unit: def register_corrector()
file: jcm/sampling.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 91
unit: def get_consistency_loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 92
unit: def likelihood_fn_repeated()
file: jcm/likelihood.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 3

id: 93
unit: def get_samples_from_ckpt()
file: jcm/metrics.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 2

id: 94
unit: def ddpm_conv1x1()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 6

id: 95
unit: def ddpm_conv3x3()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 6

id: 96
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 2

id: 97
unit: def shared_corrector_update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 9

id: 98
unit: def __init__()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 5

id: 99
unit: def flatten_dict()
file: jcm/utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 1

id: 100
unit: def denoise_update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 3

id: 101
unit: def get_continuous_consistency_loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 102
unit: def get_network()
file: jcm/models/ncsnv2.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 103
unit: def get_model_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 4

id: 104
unit: def get_act()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 105
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 3

id: 106
unit: def get_timestep_embedding()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 107
unit: def setup()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 108
unit: def get_pc_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 109
unit: def compute_fid()
file: jcm/metrics.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 9

id: 110
unit: def get_classifier_grad_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 1

id: 111
unit: def downsample_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 5

id: 112
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 113
unit: def natural_sort()
file: jcm/checkpoints.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 2

id: 114
unit: def __init__()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 4

id: 115
unit: def sde()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 116
unit: def __init__()
file: jcm/sampling.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 5

id: 117
unit: def __init__()
file: jcm/sampling.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 5

id: 118
unit: def drift_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 3

id: 119
unit: def drift_fn()
file: jcm/likelihood.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 3

id: 120
unit: def __getitem__()
file: jcm/metrics.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 2

id: 121
unit: def gaussianizer_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 3

id: 122
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 3

id: 123
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 124
unit: def latest_checkpoint()
file: jcm/checkpoints.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 125
unit: def read_chunk()
file: jcm/checkpoints.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 126
unit: def sde()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 4

id: 127
unit: def __init__()
file: jcm/sampling.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 4

id: 128
unit: def vpsde_update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 4

id: 129
unit: def get_ode_sampler()
file: jcm/sampling.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 130
unit: def euler_solver()
file: jcm/losses.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 4

id: 131
unit: def __call__()
file: jcm/models/layerspp.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 3

id: 132
unit: def _setup_kernel()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 133
unit: def _output_add()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 134
unit: def __call__()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 4

id: 135
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 136
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 137
unit: def __call__()
file: jcm/models/normalization.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 138
unit: def discretize()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 139
unit: def get_progressive_distillation_loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 140
unit: def get_likelihood_fn()
file: jcm/likelihood.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 141
unit: def __init__()
file: jcm/metrics.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 142
unit: def __init__()
file: jcm/metrics.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 143
unit: def __getitem__()
file: jcm/metrics.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 144
unit: def __call__()
file: jcm/models/layerspp.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 145
unit: def __call__()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 146
unit: def contract_inner()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 147
unit: def __call__()
file: jcm/models/normalization.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 148
unit: def __init__()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 149
unit: def bisection_func()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 150
unit: def bisection_func()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 151
unit: def marginal_prob()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 152
unit: def prior_logp()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 153
unit: def prior_entropy()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 154
unit: def __init__()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 155
unit: def prior_logp()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 156
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 4

id: 157
unit: def get_score_matching_loss_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 158
unit: def compute_all_metrics()
file: jcm/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 159
unit: def compute_metrics()
file: jcm/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 5

id: 160
unit: def get_dataset()
file: jcm/datasets.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 161
unit: def init_lpips()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 162
unit: def __call__()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 163
unit: def discretize()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 164
unit: def __init__()
file: jcm/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 5

id: 165
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 4

id: 166
unit: def compute_fid_jupyter()
file: jcm/metrics.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 167
unit: def uniform_deq()
file: jcm/datasets.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 168
unit: def data_augmentation()
file: jcm/datasets.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 169
unit: def naive_upsample_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 170
unit: def activation()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 4

id: 171
unit: def setup()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 172
unit: def setup()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 173
unit: def sde()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 174
unit: def prior_logp()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 175
unit: def prior_entropy()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 176
unit: def __init__()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 177
unit: def prior_logp()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 178
unit: def prior_entropy()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 179
unit: def prior_entropy()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 180
unit: def __init__()
file: jcm/sampling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 181
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 4

id: 182
unit: def get_step_fn()
file: jcm/losses.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 183
unit: def get_div_fn()
file: jcm/likelihood.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 184
unit: def naive_downsample_2d()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 185
unit: def dense_layer_init_fn()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 186
unit: def __call__()
file: jcm/models/ncsnpp.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 187
unit: def sde()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 188
unit: def marginal_prob()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 189
unit: def load_training_state()
file: jcm/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 190
unit: def shake_drop_eval()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 191
unit: def default_init()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 192
unit: def _einsum()
file: jcm/models/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 5

id: 193
unit: def scale_init()
file: jcm/models/normalization.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 194
unit: def scale_init()
file: jcm/models/normalization.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 195
unit: def __call__()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 196
unit: def normalize_tensor()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 197
unit: def __init__()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 198
unit: def __len__()
file: jcm/metrics.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 199
unit: def __len__()
file: jcm/metrics.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 200
unit: def obtain_feature_extractor()
file: jcm/metrics.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 201
unit: def get_model()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 202
unit: def get_distiller_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 203
unit: def get_gaussianizer_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 204
unit: def get_denoiser_and_distiller_fn()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 205
unit: def to_flattened_numpy()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 206
unit: def from_flattened_numpy()
file: jcm/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 207
unit: def get_weight()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 208
unit: def _shape()
file: jcm/models/up_or_down_sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 209
unit: def shake_shake_eval()
file: jcm/models/wideresnet_noise_conditional.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 210
unit: def spatial_average()
file: jcm/models/lpips.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 211
unit: def _checkpoint_path()
file: jcm/checkpoints.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 212
unit: def restore_checkpoint()
file: jcm/checkpoints.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 213
unit: def T()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 214
unit: def sde()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 215
unit: def marginal_prob()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 216
unit: def prior_sampling()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 217
unit: def prior_logp()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 218
unit: def T()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 219
unit: def prior_sampling()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 220
unit: def sample_importance_weighted_time_for_likelihood()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 221
unit: def T()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 222
unit: def prior_sampling()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 223
unit: def sample_importance_weighted_time_for_likelihood()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 224
unit: def T()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 225
unit: def prior_sampling()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 226
unit: def T()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 227
unit: def prior_sampling()
file: jcm/sde_lib.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 228
unit: def batch_add()
file: jcm/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 229
unit: def batch_mul()
file: jcm/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 230
unit: def get_predictor()
file: jcm/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 231
unit: def get_corrector()
file: jcm/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 232
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 233
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

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

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

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

id: 237
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

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

id: 239
unit: def update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 240
unit: def shared_predictor_update_fn()
file: jcm/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 241
unit: def get_quarter_masks()
file: jcm/losses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2