SYMBOL INDEX (92 symbols across 19 files)

FILE: fine_tune.py
  function save_progress (line 78) | def save_progress(text_encoder, placeholder_token_id, accelerator, args,...
  function parse_args (line 85) | def parse_args():
  class TextualInversionDataset (line 329) | class TextualInversionDataset(Dataset):
    method __init__ (line 330) | def __init__(
    method __len__ (line 369) | def __len__(self):
    method __getitem__ (line 372) | def __getitem__(self, i):
  function get_full_repo_name (line 412) | def get_full_repo_name(model_id: str, organization: Optional[str] = None...
  function main (line 422) | def main(args):

FILE: fine_tune_upstream.py
  function save_model_card (line 106) | def save_model_card(repo_id: str, images=None, base_model=str, repo_fold...
  function log_validation (line 134) | def log_validation(text_encoder, tokenizer, unet, vae, args, accelerator...
  function save_progress (line 180) | def save_progress(text_encoder, placeholder_token_ids, accelerator, args...
  function parse_args (line 191) | def parse_args():
  class TextualInversionDataset (line 500) | class TextualInversionDataset(Dataset):
    method __init__ (line 501) | def __init__(
    method __len__ (line 540) | def __len__(self):
    method __getitem__ (line 543) | def __getitem__(self, i):
  function main (line 586) | def main(args):

FILE: plot.py
  function pretty (line 12) | def pretty(text):
  function select_by_epoch (line 81) | def select_by_epoch(df):

FILE: plot_masking_ablation.py
  function pretty (line 15) | def pretty(text):
  function select_by_epoch (line 94) | def select_by_epoch(df):

FILE: plot_stacking_ablation.py
  function pretty (line 12) | def pretty(text):
  function select_by_epoch (line 86) | def select_by_epoch(df):

FILE: plot_stratify.py
  function pretty (line 11) | def pretty(text):
  function select_by_epoch (line 75) | def select_by_epoch(df):

FILE: semantic_aug/augmentations/compose.py
  class ComposeSequential (line 16) | class ComposeSequential(GenerativeAugmentation):
    method __init__ (line 18) | def __init__(self, augs: List[GenerativeAugmentation],
    method forward (line 27) | def forward(self, image: Image.Image, label: int,
  class ComposeParallel (line 38) | class ComposeParallel(GenerativeAugmentation):
    method __init__ (line 40) | def __init__(self, augs: List[GenerativeAugmentation],
    method forward (line 49) | def forward(self, image: Image.Image, label: int,

FILE: semantic_aug/augmentations/real_guidance.py
  class RealGuidance (line 18) | class RealGuidance(GenerativeAugmentation):
    method __init__ (line 22) | def __init__(self, model_path: str = "CompVis/stable-diffusion-v1-4",
    method forward (line 64) | def forward(self, image: Image.Image, label: int,

FILE: semantic_aug/augmentations/textual_inversion.py
  function load_embeddings (line 27) | def load_embeddings(embed_path: str,
  function format_name (line 59) | def format_name(name):
  class TextualInversion (line 63) | class TextualInversion(GenerativeAugmentation):
    method __init__ (line 67) | def __init__(self, embed_path: str,
    method forward (line 118) | def forward(self, image: Image.Image, label: int,

FILE: semantic_aug/augmentations/textual_inversion_upstream.py
  function format_name (line 27) | def format_name(name, num_tokens: int = 1):
  class TextualInversion (line 38) | class TextualInversion(GenerativeAugmentation):
    method __init__ (line 42) | def __init__(self, embed_path: str,
    method forward (line 94) | def forward(self, image: Image.Image, label: int,

FILE: semantic_aug/datasets/caltech101.py
  class CalTech101Dataset (line 19) | class CalTech101Dataset(FewShotDataset):
    method __init__ (line 43) | def __init__(self, *args, split: str = "train", seed: int = 0,
    method __len__ (line 116) | def __len__(self):
    method get_image_by_idx (line 120) | def get_image_by_idx(self, idx: int) -> Image.Image:
    method get_label_by_idx (line 124) | def get_label_by_idx(self, idx: int) -> int:
    method get_metadata_by_idx (line 128) | def get_metadata_by_idx(self, idx: int) -> dict:

FILE: semantic_aug/datasets/coco.py
  class COCODataset (line 26) | class COCODataset(FewShotDataset):
    method __init__ (line 45) | def __init__(self, *args, split: str = "train", seed: int = 0,
    method __len__ (line 139) | def __len__(self):
    method get_image_by_idx (line 143) | def get_image_by_idx(self, idx: int) -> torch.Tensor:
    method get_label_by_idx (line 147) | def get_label_by_idx(self, idx: int) -> torch.Tensor:
    method get_metadata_by_idx (line 151) | def get_metadata_by_idx(self, idx: int) -> Dict:

FILE: semantic_aug/datasets/flowers102.py
  class Flowers102Dataset (line 20) | class Flowers102Dataset(FewShotDataset):
    method __init__ (line 128) | def __init__(self, *args, split: str = "train", seed: int = 0,
    method __len__ (line 203) | def __len__(self):
    method get_image_by_idx (line 207) | def get_image_by_idx(self, idx: int) -> Image.Image:
    method get_label_by_idx (line 211) | def get_label_by_idx(self, idx: int) -> int:
    method get_metadata_by_idx (line 215) | def get_metadata_by_idx(self, idx: int) -> dict:

FILE: semantic_aug/datasets/imagenet.py
  class ImageNetDataset (line 27) | class ImageNetDataset(FewShotDataset):
    method __init__ (line 52) | def __init__(self, *args, split: str = "train", seed: int = 0,
    method __len__ (line 147) | def __len__(self):
    method get_image_by_idx (line 151) | def get_image_by_idx(self, idx: int) -> torch.Tensor:
    method get_label_by_idx (line 155) | def get_label_by_idx(self, idx: int) -> torch.Tensor:
    method get_metadata_by_idx (line 159) | def get_metadata_by_idx(self, idx: int) -> Dict:

FILE: semantic_aug/datasets/pascal.py
  class PASCALDataset (line 26) | class PASCALDataset(FewShotDataset):
    method __init__ (line 35) | def __init__(self, *args, split: str = "train", seed: int = 0,
    method __len__ (line 142) | def __len__(self):
    method get_image_by_idx (line 146) | def get_image_by_idx(self, idx: int) -> Image.Image:
    method get_label_by_idx (line 150) | def get_label_by_idx(self, idx: int) -> int:
    method get_metadata_by_idx (line 154) | def get_metadata_by_idx(self, idx: int) -> dict:

FILE: semantic_aug/datasets/spurge.py
  class SpurgeDataset (line 20) | class SpurgeDataset(FewShotDataset):
    method __init__ (line 25) | def __init__(self, *args, data_dir: str = DEFAULT_DATA_DIR,
    method __len__ (line 93) | def __len__(self):
    method get_image_by_idx (line 97) | def get_image_by_idx(self, idx: int) -> torch.Tensor:
    method get_label_by_idx (line 101) | def get_label_by_idx(self, idx: int) -> torch.Tensor:
    method get_metadata_by_idx (line 105) | def get_metadata_by_idx(self, idx: int) -> Any:

FILE: semantic_aug/few_shot_dataset.py
  class FewShotDataset (line 17) | class FewShotDataset(Dataset):
    method __init__ (line 22) | def __init__(self, examples_per_class: int = None,
    method get_image_by_idx (line 45) | def get_image_by_idx(self, idx: int) -> Image.Image:
    method get_label_by_idx (line 50) | def get_label_by_idx(self, idx: int) -> int:
    method get_metadata_by_idx (line 55) | def get_metadata_by_idx(self, idx: int) -> dict:
    method generate_augmentations (line 59) | def generate_augmentations(self, num_repeats: int):
    method __getitem__ (line 82) | def __getitem__(self, idx: int) -> Tuple[torch.Tensor, int]:

FILE: semantic_aug/generative_augmentation.py
  class GenerativeAugmentation (line 10) | class GenerativeAugmentation(nn.Module, abc.ABC):
    method forward (line 13) | def forward(self, image: Image.Image, label: int,

FILE: train_classifier.py
  function run_experiment (line 67) | def run_experiment(examples_per_class: int = 0,
  class ClassificationModel (line 320) | class ClassificationModel(nn.Module):
    method __init__ (line 322) | def __init__(self, num_classes: int, backbone: str = "resnet50"):
    method forward (line 340) | def forward(self, image):