[ {
  "shortName" : "def evaluate()",
  "longName" : "",
  "relativeFileName" : "jcm/evaluate.py",
  "fileLinesCount" : 415,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 292,
  "mcCabeIndex" : 50,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def train()",
  "longName" : "",
  "relativeFileName" : "jcm/train.py",
  "fileLinesCount" : 343,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 236,
  "mcCabeIndex" : 47,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ncsnpp.py",
  "fileLinesCount" : 271,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 180,
  "mcCabeIndex" : 36,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 136,
  "mcCabeIndex" : 8,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 127,
  "mcCabeIndex" : 10,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def step_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 83,
  "mcCabeIndex" : 6,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def dsm_loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 80,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def ssm_loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 77,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 73,
  "mcCabeIndex" : 9,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ncsnv2.py",
  "fileLinesCount" : 349,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 69,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_config()",
  "longName" : "",
  "relativeFileName" : "configs/cifar10_ve_ct_adaptive.py",
  "fileLinesCount" : 86,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 67,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ncsnv2.py",
  "fileLinesCount" : 349,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 65,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_config()",
  "longName" : "",
  "relativeFileName" : "configs/cifar10_ve_ct_ema.py",
  "fileLinesCount" : 83,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 65,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_config()",
  "longName" : "",
  "relativeFileName" : "configs/cifar10_ve_progressive_distillation.py",
  "fileLinesCount" : 83,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 65,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_config()",
  "longName" : "",
  "relativeFileName" : "configs/cifar10_ve_cd.py",
  "fileLinesCount" : 81,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 63,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 61,
  "mcCabeIndex" : 15,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_config()",
  "longName" : "",
  "relativeFileName" : "configs/cifar10_ve_continuous.py",
  "fileLinesCount" : 78,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 60,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_sampling_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 58,
  "mcCabeIndex" : 9,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_optimizer()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 58,
  "mcCabeIndex" : 6,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ncsnv2.py",
  "fileLinesCount" : 349,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 56,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_config()",
  "longName" : "",
  "relativeFileName" : "configs/cifar10_k_ve.py",
  "fileLinesCount" : 70,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 56,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_config()",
  "longName" : "",
  "relativeFileName" : "configs/cifar10_k_vp.py",
  "fileLinesCount" : 71,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 56,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_default_configs()",
  "longName" : "",
  "relativeFileName" : "configs/default_cifar10_configs.py",
  "fileLinesCount" : 73,
  "components" : [ "primary::configs" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 55,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_ema_scales_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 54,
  "mcCabeIndex" : 8,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def upfirdn_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 52,
  "mcCabeIndex" : 8,
  "numberOfParameters" : 10,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ddpm.py",
  "fileLinesCount" : 115,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 52,
  "mcCabeIndex" : 12,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ncsnv2.py",
  "fileLinesCount" : 349,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 47,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def collate_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/datasets.py",
  "fileLinesCount" : 155,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 41,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 40,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/normalization.py",
  "fileLinesCount" : 170,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 40,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_heun_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 39,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_score_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 38,
  "mcCabeIndex" : 10,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def likelihood_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/likelihood.py",
  "fileLinesCount" : 161,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 38,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_euler_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 35,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 33,
  "mcCabeIndex" : 7,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 33,
  "mcCabeIndex" : 7,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def save_image()",
  "longName" : "",
  "relativeFileName" : "jcm/utils.py",
  "fileLinesCount" : 120,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 33,
  "mcCabeIndex" : 10,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layerspp.py",
  "fileLinesCount" : 241,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 32,
  "mcCabeIndex" : 11,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 32,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_progressive_distillation_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 32,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def main()",
  "longName" : "",
  "relativeFileName" : "jcm/main.py",
  "fileLinesCount" : 86,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 32,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def upsample_conv_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 31,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 30,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_sde()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 30,
  "mcCabeIndex" : 5,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 29,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def reverse()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 28,
  "mcCabeIndex" : 7,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def ode_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 27,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def denoiser_distiller_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 26,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 26,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 25,
  "mcCabeIndex" : 6,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layerspp.py",
  "fileLinesCount" : 241,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 22,
  "mcCabeIndex" : 5,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def marginal_prob()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 22,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def marginal_prob()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 22,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 22,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layerspp.py",
  "fileLinesCount" : 241,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 21,
  "mcCabeIndex" : 5,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layerspp.py",
  "fileLinesCount" : 241,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 21,
  "mcCabeIndex" : 6,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def _simple_upfirdn_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 21,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 7,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 21,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 20,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_normalization()",
  "longName" : "",
  "relativeFileName" : "jcm/models/normalization.py",
  "fileLinesCount" : 170,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 20,
  "mcCabeIndex" : 7,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_fids()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 19,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def init_model()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 19,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_denoiser_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 19,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def conv_downsample_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 19,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def shake_drop_train()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 18,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 7,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def ncsn_conv1x1()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 18,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def ncsn_conv3x3()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 18,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 18,
  "mcCabeIndex" : 5,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def save_checkpoint()",
  "longName" : "",
  "relativeFileName" : "jcm/checkpoints.py",
  "fileLinesCount" : 215,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 18,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 17,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/normalization.py",
  "fileLinesCount" : 170,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 17,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layerspp.py",
  "fileLinesCount" : 241,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 16,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def distiller_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 16,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_logit_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 16,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 16,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def upsample_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def likelihood_importance_cum_weight()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def likelihood_importance_cum_weight()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def vesde_update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def pc_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_onestep_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_seeded_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def heun_solver()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 15,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def register_model()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def create_classifier()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def shake_shake_train()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 6,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def discretize()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def register_predictor()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def register_corrector()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_consistency_loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def likelihood_fn_repeated()",
  "longName" : "",
  "relativeFileName" : "jcm/likelihood.py",
  "fileLinesCount" : 161,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 14,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_samples_from_ckpt()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 13,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def ddpm_conv1x1()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 13,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def ddpm_conv3x3()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 13,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 13,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def shared_corrector_update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 13,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 9,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 12,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def flatten_dict()",
  "longName" : "",
  "relativeFileName" : "jcm/utils.py",
  "fileLinesCount" : 120,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 12,
  "mcCabeIndex" : 5,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def denoise_update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 12,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_continuous_consistency_loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 12,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_network()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ncsnv2.py",
  "fileLinesCount" : 349,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_model_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_act()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 5,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_timestep_embedding()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def setup()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_pc_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def compute_fid()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 9,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_classifier_grad_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def downsample_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def natural_sort()",
  "longName" : "",
  "relativeFileName" : "jcm/checkpoints.py",
  "fileLinesCount" : 215,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sde()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def drift_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def drift_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/likelihood.py",
  "fileLinesCount" : 161,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 10,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __getitem__()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 5,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def gaussianizer_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def latest_checkpoint()",
  "longName" : "",
  "relativeFileName" : "jcm/checkpoints.py",
  "fileLinesCount" : 215,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def read_chunk()",
  "longName" : "",
  "relativeFileName" : "jcm/checkpoints.py",
  "fileLinesCount" : 215,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sde()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def vpsde_update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_ode_sampler()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def euler_solver()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layerspp.py",
  "fileLinesCount" : 241,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def _setup_kernel()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def _output_add()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/normalization.py",
  "fileLinesCount" : 170,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def discretize()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_progressive_distillation_loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_likelihood_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/likelihood.py",
  "fileLinesCount" : 161,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __getitem__()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layerspp.py",
  "fileLinesCount" : 241,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def contract_inner()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/normalization.py",
  "fileLinesCount" : 170,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def bisection_func()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def bisection_func()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def marginal_prob()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_logp()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_entropy()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_logp()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_score_matching_loss_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def compute_all_metrics()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def compute_metrics()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_dataset()",
  "longName" : "",
  "relativeFileName" : "jcm/datasets.py",
  "fileLinesCount" : 155,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def init_lpips()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def discretize()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def compute_fid_jupyter()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def uniform_deq()",
  "longName" : "",
  "relativeFileName" : "jcm/datasets.py",
  "fileLinesCount" : 155,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def data_augmentation()",
  "longName" : "",
  "relativeFileName" : "jcm/datasets.py",
  "fileLinesCount" : 155,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def naive_upsample_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def activation()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def setup()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def setup()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sde()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_logp()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_entropy()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_logp()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_entropy()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_entropy()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_step_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_div_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/likelihood.py",
  "fileLinesCount" : 161,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def naive_downsample_2d()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def dense_layer_init_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/ncsnpp.py",
  "fileLinesCount" : 271,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sde()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def marginal_prob()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def load_training_state()",
  "longName" : "",
  "relativeFileName" : "jcm/utils.py",
  "fileLinesCount" : 120,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def shake_drop_eval()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def default_init()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def _einsum()",
  "longName" : "",
  "relativeFileName" : "jcm/models/layers.py",
  "fileLinesCount" : 649,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def scale_init()",
  "longName" : "",
  "relativeFileName" : "jcm/models/normalization.py",
  "fileLinesCount" : 170,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def scale_init()",
  "longName" : "",
  "relativeFileName" : "jcm/models/normalization.py",
  "fileLinesCount" : 170,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __call__()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def normalize_tensor()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __len__()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __len__()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def obtain_feature_extractor()",
  "longName" : "",
  "relativeFileName" : "jcm/metrics.py",
  "fileLinesCount" : 301,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_model()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_distiller_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_gaussianizer_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_denoiser_and_distiller_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def to_flattened_numpy()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def from_flattened_numpy()",
  "longName" : "",
  "relativeFileName" : "jcm/models/utils.py",
  "fileLinesCount" : 441,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_weight()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def _shape()",
  "longName" : "",
  "relativeFileName" : "jcm/models/up_or_down_sampling.py",
  "fileLinesCount" : 426,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def shake_shake_eval()",
  "longName" : "",
  "relativeFileName" : "jcm/models/wideresnet_noise_conditional.py",
  "fileLinesCount" : 333,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def spatial_average()",
  "longName" : "",
  "relativeFileName" : "jcm/models/lpips.py",
  "fileLinesCount" : 110,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def _checkpoint_path()",
  "longName" : "",
  "relativeFileName" : "jcm/checkpoints.py",
  "fileLinesCount" : 215,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def restore_checkpoint()",
  "longName" : "",
  "relativeFileName" : "jcm/checkpoints.py",
  "fileLinesCount" : 215,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def T()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sde()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def marginal_prob()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_sampling()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_logp()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def T()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_sampling()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sample_importance_weighted_time_for_likelihood()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def T()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_sampling()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sample_importance_weighted_time_for_likelihood()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def T()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_sampling()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def T()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def prior_sampling()",
  "longName" : "",
  "relativeFileName" : "jcm/sde_lib.py",
  "fileLinesCount" : 511,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 3,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def batch_add()",
  "longName" : "",
  "relativeFileName" : "jcm/utils.py",
  "fileLinesCount" : 120,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def batch_mul()",
  "longName" : "",
  "relativeFileName" : "jcm/utils.py",
  "fileLinesCount" : 120,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_predictor()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_corrector()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def __init__()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def shared_predictor_update_fn()",
  "longName" : "",
  "relativeFileName" : "jcm/sampling.py",
  "fileLinesCount" : 785,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def get_quarter_masks()",
  "longName" : "",
  "relativeFileName" : "jcm/losses.py",
  "fileLinesCount" : 1078,
  "components" : [ "primary::jcm" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
} ]