SYMBOL INDEX (28 symbols across 1 files)

FILE: ResViT.py
  function _weights_init (line 15) | def _weights_init(m):
  class LambdaLayer (line 21) | class LambdaLayer(nn.Module):
    method __init__ (line 22) | def __init__(self, lambd):
    method forward (line 26) | def forward(self, x):
  class BasicBlock (line 30) | class BasicBlock(nn.Module):
    method __init__ (line 33) | def __init__(self, in_planes, planes, stride=1, option='A'):
    method forward (line 54) | def forward(self, x):
  class Residual (line 64) | class Residual(nn.Module):
    method __init__ (line 65) | def __init__(self, fn):
    method forward (line 68) | def forward(self, x, **kwargs):
  class LayerNormalize (line 71) | class LayerNormalize(nn.Module):
    method __init__ (line 72) | def __init__(self, dim, fn):
    method forward (line 76) | def forward(self, x, **kwargs):
  class MLP_Block (line 79) | class MLP_Block(nn.Module):
    method __init__ (line 80) | def __init__(self, dim, hidden_dim, dropout = 0.1):
    method forward (line 92) | def forward(self, x):
  class Attention (line 101) | class Attention(nn.Module):
    method __init__ (line 102) | def __init__(self, dim, heads = 8, dropout = 0.1):
    method forward (line 117) | def forward(self, x, mask = None):
  class Transformer (line 139) | class Transformer(nn.Module):
    method __init__ (line 140) | def __init__(self, dim, depth, heads, mlp_dim, dropout):
    method forward (line 148) | def forward(self, x, mask = None):
  class ViTResNet (line 155) | class ViTResNet(nn.Module):
    method __init__ (line 156) | def __init__(self, block, num_blocks, num_classes=10, dim = 128, num_t...
    method _make_layer (line 194) | def _make_layer(self, block, planes, num_blocks, stride):
    method forward (line 205) | def forward(self, img, mask = None):
  function train (line 260) | def train(model, optimizer, data_loader, loss_history):
  function evaluate (line 277) | def evaluate(model, data_loader, loss_history):