SYMBOL INDEX (135 symbols across 10 files)

FILE: checkpoint.py
  function setup_arg_parser (line 10) | def setup_arg_parser(parser: argparse.ArgumentParser):
  function _find_autoresume_path (line 22) | def _find_autoresume_path(args: argparse.Namespace):
  function resume_from_checkpoint (line 46) | def resume_from_checkpoint(
  function save_checkpoint (line 79) | def save_checkpoint(

FILE: main.py
  function setup_print (line 17) | def setup_print(is_master: bool):
  function main (line 33) | def main():
  function evaluate (line 199) | def evaluate(model: torch.nn.Module, loader: torch.utils.data.DataLoader):

FILE: model.py
  class TemporalCrossAttention (line 18) | class TemporalCrossAttention(nn.Module):
    method __init__ (line 20) | def __init__(
    method forward_half (line 44) | def forward_half(self, q: torch.Tensor, k: torch.Tensor, w: torch.Tens...
    method forward (line 60) | def forward(self, q: torch.Tensor, k: torch.Tensor):
  class EVLDecoder (line 71) | class EVLDecoder(nn.Module):
    method __init__ (line 73) | def __init__(
    method _initialize_weights (line 114) | def _initialize_weights(self):
    method forward (line 118) | def forward(self, in_features: List[Dict[str, torch.Tensor]]):
  class EVLTransformer (line 146) | class EVLTransformer(nn.Module):
    method __init__ (line 148) | def __init__(
    method _create_backbone (line 194) | def _create_backbone(
    method _get_backbone (line 220) | def _get_backbone(self, x):
    method forward (line 230) | def forward(self, x: torch.Tensor):

FILE: video_dataset/dataloader.py
  function setup_arg_parser (line 11) | def setup_arg_parser(parser: argparse.ArgumentParser):
  function _parse_mean_and_std (line 59) | def _parse_mean_and_std(args: argparse.Namespace) -> Dict[str, torch.Ten...
  function create_train_dataset (line 75) | def create_train_dataset(args: argparse.Namespace) -> torch.utils.data.D...
  function create_train_loader (line 98) | def create_train_loader(args: argparse.Namespace, resume_step: int = 0) ...
  function create_val_dataset (line 123) | def create_val_dataset(args: argparse.Namespace) -> torch.utils.data.Dat...
  function create_val_loader (line 145) | def create_val_loader(args: argparse.Namespace) -> torch.utils.data.Data...

FILE: video_dataset/dataset.py
  class VideoDataset (line 14) | class VideoDataset(torch.utils.data.Dataset):
    method __init__ (line 16) | def __init__(
    method __len__ (line 46) | def __len__(self):
    method __getitem__ (line 50) | def __getitem__(self, idx):
    method _generate_temporal_crops (line 113) | def _generate_temporal_crops(self, frames):
    method _generate_spatial_crops (line 131) | def _generate_spatial_crops(self, frames):
    method _random_sample_frame_idx (line 155) | def _random_sample_frame_idx(self, len):
  class DummyDataset (line 173) | class DummyDataset(torch.utils.data.Dataset):
    method __init__ (line 175) | def __init__(self, list_path: str, num_frames: int, num_views: int, sp...
    method __len__ (line 182) | def __len__(self):
    method __getitem__ (line 185) | def __getitem__(self, _):

FILE: video_dataset/rand_augment.py
  function _interpolation (line 55) | def _interpolation(kwargs):
  function _check_args_tf (line 63) | def _check_args_tf(kwargs):
  function shear_x (line 69) | def shear_x(img, factor, **kwargs):
  function shear_y (line 76) | def shear_y(img, factor, **kwargs):
  function translate_x_rel (line 83) | def translate_x_rel(img, pct, **kwargs):
  function translate_y_rel (line 91) | def translate_y_rel(img, pct, **kwargs):
  function translate_x_abs (line 99) | def translate_x_abs(img, pixels, **kwargs):
  function translate_y_abs (line 106) | def translate_y_abs(img, pixels, **kwargs):
  function rotate (line 113) | def rotate(img, degrees, **kwargs):
  function auto_contrast (line 147) | def auto_contrast(img, **__):
  function invert (line 151) | def invert(img, **__):
  function equalize (line 155) | def equalize(img, **__):
  function solarize (line 159) | def solarize(img, thresh, **__):
  function solarize_add (line 163) | def solarize_add(img, add, thresh=128, **__):
  function posterize (line 178) | def posterize(img, bits_to_keep, **__):
  function contrast (line 184) | def contrast(img, factor, **__):
  function color (line 188) | def color(img, factor, **__):
  function brightness (line 192) | def brightness(img, factor, **__):
  function sharpness (line 196) | def sharpness(img, factor, **__):
  function _randomly_negate (line 200) | def _randomly_negate(v):
  function _rotate_level_to_arg (line 205) | def _rotate_level_to_arg(level, _hparams):
  function _enhance_level_to_arg (line 212) | def _enhance_level_to_arg(level, _hparams):
  function _enhance_increasing_level_to_arg (line 217) | def _enhance_increasing_level_to_arg(level, _hparams):
  function _shear_level_to_arg (line 225) | def _shear_level_to_arg(level, _hparams):
  function _translate_abs_level_to_arg (line 232) | def _translate_abs_level_to_arg(level, hparams):
  function _translate_rel_level_to_arg (line 239) | def _translate_rel_level_to_arg(level, hparams):
  function _posterize_level_to_arg (line 247) | def _posterize_level_to_arg(level, _hparams):
  function _posterize_increasing_level_to_arg (line 254) | def _posterize_increasing_level_to_arg(level, hparams):
  function _posterize_original_level_to_arg (line 261) | def _posterize_original_level_to_arg(level, _hparams):
  function _solarize_level_to_arg (line 268) | def _solarize_level_to_arg(level, _hparams):
  function _solarize_increasing_level_to_arg (line 274) | def _solarize_increasing_level_to_arg(level, _hparams):
  function _solarize_add_level_to_arg (line 280) | def _solarize_add_level_to_arg(level, _hparams):
  class AugmentOp (line 342) | class AugmentOp:
    method __init__ (line 347) | def __init__(self, name, prob=0.5, magnitude=10, hparams=None):
    method __call__ (line 369) | def __call__(self, img_list):
  function _select_rand_weights (line 449) | def _select_rand_weights(weight_idx=0, transforms=None):
  function rand_augment_ops (line 458) | def rand_augment_ops(magnitude=10, hparams=None, transforms=None):
  class RandAugment (line 467) | class RandAugment:
    method __init__ (line 468) | def __init__(self, ops, num_layers=2, choice_weights=None):
    method __call__ (line 473) | def __call__(self, img):
  function rand_augment_transform (line 486) | def rand_augment_transform(config_str, hparams):

FILE: video_dataset/random_erasing.py
  function _get_pixels (line 21) | def _get_pixels(
  class RandomErasing (line 37) | class RandomErasing:
    method __init__ (line 56) | def __init__(
    method _erase (line 90) | def _erase(self, img, chan, img_h, img_w, dtype):
    method _erase_cube (line 119) | def _erase_cube(
    method __call__ (line 161) | def __call__(self, input):

FILE: video_dataset/transform.py
  function _pil_interp (line 33) | def _pil_interp(method):
  function random_short_side_scale_jitter (line 47) | def random_short_side_scale_jitter(
  function crop_boxes (line 104) | def crop_boxes(boxes, x_offset, y_offset):
  function random_crop (line 123) | def random_crop(images, size, boxes=None):
  function horizontal_flip (line 159) | def horizontal_flip(prob, images, boxes=None):
  function uniform_crop (line 194) | def uniform_crop(images, size, spatial_idx, boxes=None, scale_size=None):
  function clip_boxes_to_image (line 257) | def clip_boxes_to_image(boxes, height, width):
  function blend (line 279) | def blend(images1, images2, alpha):
  function grayscale (line 295) | def grayscale(images):
  function color_jitter (line 317) | def color_jitter(images, img_brightness=0, img_contrast=0, img_saturatio...
  function brightness_jitter (line 352) | def brightness_jitter(var, images):
  function contrast_jitter (line 371) | def contrast_jitter(var, images):
  function saturation_jitter (line 391) | def saturation_jitter(var, images):
  function lighting_jitter (line 410) | def lighting_jitter(images, alphastd, eigval, eigvec):
  function color_normalization (line 458) | def color_normalization(images, mean, stddev):
  function _get_param_spatial_crop (line 502) | def _get_param_spatial_crop(
  function random_resized_crop (line 544) | def random_resized_crop(
  function random_resized_crop_with_shift (line 579) | def random_resized_crop_with_shift(
  function create_random_augment (line 624) | def create_random_augment(
  function random_sized_crop_img (line 660) | def random_sized_crop_img(
  class RandomResizedCropAndInterpolation (line 695) | class RandomResizedCropAndInterpolation:
    method __init__ (line 708) | def __init__(
    method get_params (line 730) | def get_params(img, scale, ratio):
    method __call__ (line 770) | def __call__(self, img):
    method __repr__ (line 784) | def __repr__(self):

FILE: vision_transformer.py
  class QuickGELU (line 15) | class QuickGELU(nn.Module):
    method forward (line 16) | def forward(self, x: torch.Tensor):
  class LayerNorm (line 19) | class LayerNorm(nn.LayerNorm):
    method forward (line 22) | def forward(self, x: torch.Tensor):
  class Attention (line 28) | class Attention(nn.Module):
    method __init__ (line 33) | def __init__(
    method _initialize_weights (line 52) | def _initialize_weights(self):
    method forward (line 58) | def forward(self, q: torch.Tensor, k: torch.Tensor, v: torch.Tensor):
  class PatchEmbed2D (line 84) | class PatchEmbed2D(nn.Module):
    method __init__ (line 86) | def __init__(
    method _initialize_weights (line 100) | def _initialize_weights(self, x):
    method forward (line 105) | def forward(self, x: torch.Tensor):
  class TransformerEncoderLayer (line 116) | class TransformerEncoderLayer(nn.Module):
    method __init__ (line 118) | def __init__(
    method _initialize_weights (line 152) | def _initialize_weights(self):
    method forward (line 158) | def forward(self, x: torch.Tensor):
  class TransformerDecoderLayer (line 183) | class TransformerDecoderLayer(nn.Module):
    method __init__ (line 185) | def __init__(
    method _initialize_weights (line 216) | def _initialize_weights(self):
    method forward (line 222) | def forward(self, x: torch.Tensor, y: torch.Tensor):
  class VisionTransformer2D (line 230) | class VisionTransformer2D(nn.Module):
    method __init__ (line 232) | def __init__(
    method _initialize_weights (line 269) | def _initialize_weights(self):
    method forward (line 273) | def forward(self, x: torch.Tensor):
  function model_to_fp16 (line 297) | def model_to_fp16(model: VisionTransformer2D):

FILE: weight_loaders.py
  function load_weights_clip (line 10) | def load_weights_clip(load_path: str) -> Dict[str, torch.Tensor]: