SYMBOL INDEX (39 symbols across 7 files)

FILE: learning_rate_finder.py
  function parse_config (line 10) | def parse_config():

FILE: swa.py
  function swa (line 11) | def swa(paths):
  function parse_config (line 23) | def parse_config():

FILE: tools/datasets.py
  class SupConDatasetCifar10 (line 3) | class SupConDatasetCifar10(torchvision.datasets.CIFAR10):
    method __init__ (line 4) | def __init__(self, data_dir, train, transform, second_stage):
    method __getitem__ (line 10) | def __getitem__(self, idx):
  class SupConDatasetCifar100 (line 25) | class SupConDatasetCifar100(torchvision.datasets.CIFAR100):
    method __init__ (line 26) | def __init__(self, data_dir, train, transform, second_stage):
    method __getitem__ (line 32) | def __getitem__(self, idx):
  function create_supcon_dataset (line 51) | def create_supcon_dataset(dataset_name, data_dir, train, transform, seco...

FILE: tools/losses.py
  class SupConLoss (line 5) | class SupConLoss(nn.Module):
    method __init__ (line 8) | def __init__(self, temperature=0.07, contrast_mode='all',
    method forward (line 15) | def forward(self, features, labels=None, mask=None):
  class LabelSmoothingLoss (line 95) | class LabelSmoothingLoss(nn.Module):
    method __init__ (line 96) | def __init__(self, classes, smoothing=0, dim=-1):
    method forward (line 103) | def forward(self, pred, target):

FILE: tools/models.py
  function create_encoder (line 9) | def create_encoder(backbone):
  class SupConModel (line 34) | class SupConModel(nn.Module):
    method __init__ (line 35) | def __init__(self, backbone='resnet50', projection_dim=128, second_sta...
    method use_projection_head (line 53) | def use_projection_head(self, mode):
    method forward (line 60) | def forward(self, x):

FILE: tools/utils.py
  function seed_everything (line 20) | def seed_everything(seed=42):
  function add_to_logs (line 29) | def add_to_logs(logging, message):
  function add_to_tensorboard_logs (line 33) | def add_to_tensorboard_logs(writer, message, tag, index):
  class TwoCropTransform (line 37) | class TwoCropTransform:
    method __init__ (line 39) | def __init__(self, crop_transform):
    method __call__ (line 42) | def __call__(self, x):
  function build_transforms (line 46) | def build_transforms(second_stage):
  function build_loaders (line 81) | def build_loaders(data_dir, transforms, batch_sizes, num_workers, second...
  function build_model (line 114) | def build_model(backbone, second_stage=False, num_classes=None, ckpt_pre...
  function build_optim (line 123) | def build_optim(model, optimizer_params, scheduler_params, loss_params):
  function compute_embeddings (line 139) | def compute_embeddings(loader, model, scaler):
  function train_epoch_constructive (line 163) | def train_epoch_constructive(train_loader, model, criterion, optimizer, ...
  function validation_constructive (line 206) | def validation_constructive(valid_loader, train_loader, model, scaler):
  function train_epoch_ce (line 227) | def train_epoch_ce(train_loader, model, criterion, optimizer, scaler, ema):
  function validation_ce (line 258) | def validation_ce(model, criterion, valid_loader, scaler):
  function copy_parameters_from_model (line 299) | def copy_parameters_from_model(model):
  function copy_parameters_to_model (line 304) | def copy_parameters_to_model(copy_of_model_parameters, model):

FILE: train.py
  function parse_config (line 16) | def parse_config():