SYMBOL INDEX (76 symbols across 7 files)

FILE: vit/src/ViT.py
  class PatchEmbeddings (line 3) | class PatchEmbeddings(nn.Module):
    method __init__ (line 8) | def __init__(self, config):
    method forward (line 17) | def forward(self, x):
  class Embeddings (line 23) | class Embeddings(nn.Module):
    method __init__ (line 29) | def __init__(self,config):
    method forward (line 39) | def forward(self,x):
  class Attention (line 53) | class Attention(nn.Module):
    method __init__ (line 60) | def __init__(self,vector_dim,attention_head_size,dropout,bias = True):
    method forward (line 72) | def forward(self,x):
  class MultiheadAttention (line 84) | class MultiheadAttention(nn.Module):
    method __init__ (line 92) | def __init__(self,config):
    method forward (line 116) | def forward(self,x,output_attentions = False):
  class MLP (line 129) | class MLP(nn.Module):
    method __init__ (line 134) | def __init__(self,config):
    method forward (line 141) | def forward(self,x):
  class Block (line 148) | class Block(nn.Module):
    method __init__ (line 154) | def __init__(self,config):
    method forward (line 161) | def forward(self,x,output_attentions = False):
  class Encoder (line 172) | class Encoder(nn.Module):
    method __init__ (line 174) | def __init__(self,config):
    method forward (line 182) | def forward(self, x, output_attentions=False):
  class Classification (line 195) | class Classification(nn.Module):
    method __init__ (line 201) | def __init__(self,config):
    method forward (line 215) | def forward(self, x, output_attentions=False):
    method _init_weights (line 228) | def _init_weights(self, module):

FILE: vit/src/base.py
  class NewGELUActivation (line 15) | class NewGELUActivation(nn.Module):
    method forward (line 23) | def forward(self, input):

FILE: vit/src/data.py
  function prepare_data (line 7) | def prepare_data(batch_size=4, num_workers=2, train_sample_size=None, te...

FILE: vit/src/trainer.py
  class Trainer (line 28) | class Trainer:
    method __init__ (line 33) | def __init__(self,model,optimizer,loss_fn,exp_name,device):
    method train (line 40) | def train(self,train_loader,test_loader,epochs,save_exp_every_n_epochs...
    method train_epoch (line 57) | def train_epoch(self,train_loader):
    method evaluate (line 75) | def evaluate(self,test_loader):
  function parse_args (line 99) | def parse_args():
  function main (line 115) | def main():

FILE: vit/src/utils.py
  function save_experiment (line 11) | def save_experiment(experiment_name,config,model,train_losses,test_losse...
  function save_checkpoint (line 30) | def save_checkpoint(experiment_name, model, epoch, base_dir="experiments"):
  function load_experiment (line 36) | def load_experiment(experiment_name,checkpoint_name="model_final.pt",bas...

FILE: vit/src/vit_with_lora.py
  class LoRALayer (line 7) | class LoRALayer(nn.Module):
    method __init__ (line 9) | def __init__(self, in_features, out_features, rank=4, alpha=16):
    method forward (line 19) | def forward(self, x):
  class PatchEmbeddings (line 23) | class PatchEmbeddings(nn.Module):
    method __init__ (line 28) | def __init__(self, config):
    method forward (line 37) | def forward(self, x):
  class Embeddings (line 43) | class Embeddings(nn.Module):
    method __init__ (line 49) | def __init__(self,config):
    method forward (line 56) | def forward(self,x):
  class Attention (line 70) | class Attention(nn.Module):
    method __init__ (line 74) | def __init__(self,vector_dim,attention_head_size,dropout,bias=True, us...
    method forward (line 88) | def forward(self, x):
  class MultiheadAttention (line 104) | class MultiheadAttention(nn.Module):
    method __init__ (line 108) | def __init__(self, config):
    method forward (line 137) | def forward(self, x, output_attentions=False):
  class MLP (line 156) | class MLP(nn.Module):
    method __init__ (line 161) | def __init__(self, config):
    method forward (line 186) | def forward(self, x):
  function prepare_mlp_for_lora_training (line 197) | def prepare_mlp_for_lora_training(model):
  class Block (line 206) | class Block(nn.Module):
    method __init__ (line 210) | def __init__(self, config):
    method forward (line 217) | def forward(self, x, output_attentions = False):
  class Encoder (line 227) | class Encoder(nn.Module):
    method __init__ (line 232) | def __init__(self, config):
    method forward (line 238) | def forward(self, x, output_attentions=False):
  class LoRALinear (line 251) | class LoRALinear(nn.Module):
    method __init__ (line 255) | def __init__(self, in_features, out_features, rank=8, alpha=16):
    method forward (line 260) | def forward(self, x):
  class Classification (line 263) | class Classification(nn.Module):
    method __init__ (line 268) | def __init__(self, config):
    method forward (line 292) | def forward(self, x, output_attentions=False):
    method _init_weights (line 307) | def _init_weights(self, module):
  function prepare_model_for_lora_training (line 326) | def prepare_model_for_lora_training(model):

FILE: vit/visualize/vis.py
  class PatchEmbedding (line 7) | class PatchEmbedding(nn.Module):
    method __init__ (line 8) | def __init__(self, num_patches, vector_dim, patch_size):
    method forward (line 12) | def forward(self, x):
  function visualize_patches (line 25) | def visualize_patches(input_image, patch_size):
  function visualize_feature_maps (line 44) | def visualize_feature_maps(feature_maps, num_maps_to_show=8):