SYMBOL INDEX (78 symbols across 7 files)

FILE: main.py
  function get_args_parser (line 21) | def get_args_parser():
  function main (line 176) | def main(args):

FILE: models/swiftformer.py
  function stem (line 28) | def stem(in_chs, out_chs):
  class Embedding (line 42) | class Embedding(nn.Module):
    method __init__ (line 49) | def __init__(self, patch_size=16, stride=16, padding=0,
    method forward (line 59) | def forward(self, x):
  class ConvEncoder (line 65) | class ConvEncoder(nn.Module):
    method __init__ (line 72) | def __init__(self, dim, hidden_dim=64, kernel_size=3, drop_path=0., us...
    method _init_weights (line 86) | def _init_weights(self, m):
    method forward (line 92) | def forward(self, x):
  class Mlp (line 106) | class Mlp(nn.Module):
    method __init__ (line 113) | def __init__(self, in_features, hidden_features=None,
    method _init_weights (line 125) | def _init_weights(self, m):
    method forward (line 131) | def forward(self, x):
  class EfficientAdditiveAttnetion (line 141) | class EfficientAdditiveAttnetion(nn.Module):
    method __init__ (line 148) | def __init__(self, in_dims=512, token_dim=256, num_heads=2):
    method forward (line 159) | def forward(self, x):
  class SwiftFormerLocalRepresentation (line 184) | class SwiftFormerLocalRepresentation(nn.Module):
    method __init__ (line 191) | def __init__(self, dim, kernel_size=3, drop_path=0., use_layer_scale=T...
    method _init_weights (line 205) | def _init_weights(self, m):
    method forward (line 211) | def forward(self, x):
  class SwiftFormerEncoder (line 225) | class SwiftFormerEncoder(nn.Module):
    method __init__ (line 232) | def __init__(self, dim, mlp_ratio=4.,
    method forward (line 252) | def forward(self, x):
  function Stage (line 268) | def Stage(dim, index, layers, mlp_ratio=4.,
  class SwiftFormer (line 296) | class SwiftFormer(nn.Module):
    method __init__ (line 298) | def __init__(self, layers, embed_dims=None,
    method init_weights (line 376) | def init_weights(self, pretrained=None):
    method _init_weights (line 406) | def _init_weights(self, m):
    method forward_tokens (line 415) | def forward_tokens(self, x):
    method forward (line 427) | def forward(self, x):
  function _cfg (line 445) | def _cfg(url='', **kwargs):
  function SwiftFormer_XS (line 457) | def SwiftFormer_XS(pretrained=False, **kwargs):
  function SwiftFormer_S (line 469) | def SwiftFormer_S(pretrained=False, **kwargs):
  function SwiftFormer_L1 (line 481) | def SwiftFormer_L1(pretrained=False, **kwargs):
  function SwiftFormer_L3 (line 493) | def SwiftFormer_L3(pretrained=False, **kwargs):

FILE: util/datasets.py
  class INatDataset (line 12) | class INatDataset(ImageFolder):
    method __init__ (line 13) | def __init__(self, root, train=True, year=2018, transform=None, target...
  function build_dataset (line 57) | def build_dataset(is_train, args):
  function build_transform (line 89) | def build_transform(is_train, args):

FILE: util/engine.py
  function train_one_epoch (line 17) | def train_one_epoch(model: torch.nn.Module, criterion: DistillationLoss,
  function evaluate (line 70) | def evaluate(data_loader, model, device):

FILE: util/losses.py
  class DistillationLoss (line 8) | class DistillationLoss(torch.nn.Module):
    method __init__ (line 14) | def __init__(self, base_criterion: torch.nn.Module, teacher_model: tor...
    method forward (line 24) | def forward(self, inputs, outputs, labels):

FILE: util/samplers.py
  class RASampler (line 6) | class RASampler(torch.utils.data.Sampler):
    method __init__ (line 14) | def __init__(self, dataset, num_replicas=None, rank=None, shuffle=True):
    method __iter__ (line 36) | def __iter__(self):
    method __len__ (line 56) | def __len__(self):
    method set_epoch (line 59) | def set_epoch(self, epoch):

FILE: util/utils.py
  class SmoothedValue (line 17) | class SmoothedValue(object):
    method __init__ (line 22) | def __init__(self, window_size=20, fmt=None):
    method update (line 30) | def update(self, value, n=1):
    method synchronize_between_processes (line 35) | def synchronize_between_processes(self):
    method median (line 50) | def median(self):
    method avg (line 55) | def avg(self):
    method global_avg (line 60) | def global_avg(self):
    method max (line 64) | def max(self):
    method value (line 68) | def value(self):
    method __str__ (line 71) | def __str__(self):
  class MetricLogger (line 80) | class MetricLogger(object):
    method __init__ (line 81) | def __init__(self, delimiter="\t"):
    method update (line 85) | def update(self, **kwargs):
    method __getattr__ (line 92) | def __getattr__(self, attr):
    method __str__ (line 100) | def __str__(self):
    method synchronize_between_processes (line 108) | def synchronize_between_processes(self):
    method add_meter (line 112) | def add_meter(self, name, meter):
    method log_every (line 115) | def log_every(self, iterable, print_freq, header=None):
  function _load_checkpoint_for_ema (line 162) | def _load_checkpoint_for_ema(model_ema, checkpoint):
  function setup_for_distributed (line 172) | def setup_for_distributed(is_master):
  function is_dist_avail_and_initialized (line 187) | def is_dist_avail_and_initialized():
  function get_world_size (line 195) | def get_world_size():
  function get_rank (line 201) | def get_rank():
  function is_main_process (line 207) | def is_main_process():
  function save_on_master (line 211) | def save_on_master(*args, **kwargs):
  function init_distributed_mode (line 216) | def init_distributed_mode(args):
  function replace_batchnorm (line 261) | def replace_batchnorm(net):
  function replace_layernorm (line 273) | def replace_layernorm(net):