SYMBOL INDEX (67 symbols across 11 files)

FILE: clip_sample.py
  function make_normalize (line 24) | def make_normalize(mean, std):
  function norm2 (line 33) | def norm2(x):
  function spherical_dist_loss (line 38) | def spherical_dist_loss(x, y):
  function main (line 43) | def main():

FILE: diffusion/models/danbooru_128.py
  class FourierFeatures (line 8) | class FourierFeatures(hk.Module):
    method __init__ (line 9) | def __init__(self, output_size, std=1., name=None):
    method __call__ (line 15) | def __call__(self, x):
  class Dropout2d (line 22) | class Dropout2d(hk.Module):
    method __init__ (line 23) | def __init__(self, rate=0.5, name=None):
    method __call__ (line 27) | def __call__(self, x, enabled):
  class SelfAttention2d (line 34) | class SelfAttention2d(hk.Module):
    method __init__ (line 35) | def __init__(self, n_head=1, dropout_rate=0.1, name=None):
    method __call__ (line 40) | def __call__(self, x, dropout_enabled):
  function res_conv_block (line 55) | def res_conv_block(c_mid, c_out, dropout_last=True):
  function diffusion_model (line 70) | def diffusion_model(x, t, extra_args):
  class Danbooru128Model (line 131) | class Danbooru128Model:

FILE: diffusion/models/imagenet_128.py
  class FourierFeatures (line 8) | class FourierFeatures(hk.Module):
    method __init__ (line 9) | def __init__(self, output_size, std=1., name=None):
    method __call__ (line 15) | def __call__(self, x):
  class Dropout2d (line 22) | class Dropout2d(hk.Module):
    method __init__ (line 23) | def __init__(self, rate=0.5, name=None):
    method __call__ (line 27) | def __call__(self, x, enabled):
  class SelfAttention2d (line 34) | class SelfAttention2d(hk.Module):
    method __init__ (line 35) | def __init__(self, c_in, n_head=1, dropout_rate=0.1, name=None):
    method __call__ (line 42) | def __call__(self, x, dropout_enabled):
  function res_conv_block (line 56) | def res_conv_block(c_mid, c_out, dropout_last=True):
  function diffusion_model (line 72) | def diffusion_model(x, t, extra_args):
  class ImageNet128Model (line 169) | class ImageNet128Model:

FILE: diffusion/models/models.py
  function get_model (line 17) | def get_model(model):
  function get_models (line 21) | def get_models():
  function load_params (line 25) | def load_params(checkpoint):

FILE: diffusion/models/wikiart_128.py
  class FourierFeatures (line 8) | class FourierFeatures(hk.Module):
    method __init__ (line 9) | def __init__(self, output_size, std=1., name=None):
    method __call__ (line 15) | def __call__(self, x):
  class Dropout2d (line 22) | class Dropout2d(hk.Module):
    method __init__ (line 23) | def __init__(self, rate=0.5, name=None):
    method __call__ (line 27) | def __call__(self, x, enabled):
  function res_conv_block (line 34) | def res_conv_block(c_mid, c_out, dropout_last=True):
  function diffusion_model (line 50) | def diffusion_model(x, t, extra_args):
  class WikiArt128Model (line 123) | class WikiArt128Model:

FILE: diffusion/models/wikiart_256.py
  class FourierFeatures (line 8) | class FourierFeatures(hk.Module):
    method __init__ (line 9) | def __init__(self, output_size, std=1., name=None):
    method __call__ (line 15) | def __call__(self, x):
  class Dropout2d (line 22) | class Dropout2d(hk.Module):
    method __init__ (line 23) | def __init__(self, rate=0.5, name=None):
    method __call__ (line 27) | def __call__(self, x, enabled):
  class SelfAttention2d (line 34) | class SelfAttention2d(hk.Module):
    method __init__ (line 35) | def __init__(self, n_head=1, dropout_rate=0.1, name=None):
    method __call__ (line 40) | def __call__(self, x, dropout_enabled):
  function res_conv_block (line 55) | def res_conv_block(c_mid, c_out, dropout_last=True):
  function diffusion_model (line 71) | def diffusion_model(x, t, extra_args):
  class WikiArt256Model (line 179) | class WikiArt256Model:

FILE: diffusion/sampling.py
  function sample_step (line 9) | def sample_step(model, params, key, x, t, t_next, eta, extra_args):
  function cond_sample_step (line 28) | def cond_sample_step(model, params, key, x, t, t_next, eta, extra_args, ...
  function sample_loop (line 49) | def sample_loop(model, params, key, x, steps, eta, sample_step):
  function reverse_sample_step (line 59) | def reverse_sample_step(model, params, key, x, t, t_next, extra_args):
  function reverse_sample_loop (line 73) | def reverse_sample_loop(model, params, key, x, steps, sample_step):

FILE: diffusion/utils.py
  function from_pil_image (line 7) | def from_pil_image(x):
  function to_pil_image (line 15) | def to_pil_image(x):
  function log_snr_to_alpha_sigma (line 28) | def log_snr_to_alpha_sigma(log_snr):
  function alpha_sigma_to_log_snr (line 34) | def alpha_sigma_to_log_snr(alpha, sigma):
  function t_to_alpha_sigma (line 40) | def t_to_alpha_sigma(t):
  function alpha_sigma_to_t (line 46) | def alpha_sigma_to_t(alpha, sigma):
  function get_ddpm_schedule (line 52) | def get_ddpm_schedule(ddpm_t):

FILE: interpolate.py
  function main (line 19) | def main():

FILE: make_grid.py
  function main (line 12) | def main():

FILE: sample.py
  function main (line 19) | def main():