SYMBOL INDEX (32 symbols across 3 files)

FILE: main/diffusion_module.py
  function dropout (line 34) | def dropout(proba: float):
  class Model (line 37) | class Model(pl.LightningModule):
    method __init__ (line 38) | def __init__(
    method device (line 59) | def device(self):
    method configure_optimizers (line 62) | def configure_optimizers(self):
    method training_step (line 72) | def training_step(self, batch, batch_idx):
    method validation_step (line 82) | def validation_step(self, batch, batch_idx):
  class Datamodule (line 91) | class Datamodule(pl.LightningDataModule):
    method __init__ (line 92) | def __init__(
    method setup (line 111) | def setup(self, stage: Any = None) -> None:
    method get_dataloader (line 115) | def get_dataloader(self, dataset) -> DataLoader:
    method train_dataloader (line 125) | def train_dataloader(self) -> DataLoader:
    method val_dataloader (line 128) | def val_dataloader(self) -> DataLoader:
  function get_wandb_logger (line 134) | def get_wandb_logger(trainer: Trainer) -> Optional[WandbLogger]:
  function log_wandb_audio_batch (line 149) | def log_wandb_audio_batch(
  function log_wandb_audio_spectrogram (line 166) | def log_wandb_audio_spectrogram(
  class SampleLogger (line 200) | class SampleLogger(Callback):
    method __init__ (line 201) | def __init__(
    method on_validation_epoch_start (line 219) | def on_validation_epoch_start(self, trainer, pl_module):
    method on_validation_batch_start (line 222) | def on_validation_batch_start(
    method log_sample (line 230) | def log_sample(self, trainer, pl_module, batch):

FILE: main/utils.py
  function get_logger (line 17) | def get_logger(name=__name__) -> logging.Logger:
  function extras (line 41) | def extras(config: DictConfig) -> None:
  function print_config (line 60) | def print_config(
  function log_hyperparameters (line 110) | def log_hyperparameters(
  function finish (line 154) | def finish(
  function get_packages_list (line 172) | def get_packages_list() -> List[str]:
  function retry_if_error (line 176) | def retry_if_error(fn: Callable, num_attemps: int = 10):
  class SavePytorchModelAndStopCallback (line 186) | class SavePytorchModelAndStopCallback(Callback):
    method __init__ (line 187) | def __init__(self, path: str, attribute: Optional[str] = None):
    method on_train_start (line 191) | def on_train_start(self, trainer, pl_module):

FILE: train.py
  function main (line 16) | def main(config: DictConfig) -> None: