SYMBOL INDEX (139 symbols across 11 files)

FILE: examples/rtx1_example.py
  function run (line 5) | def run(pretrained=False):

FILE: examples/train_example.py
  function run (line 5) | def run(

FILE: rtx/data_util.py
  function map_np (line 10) | def map_np(input: np.ndarray, idxs: list[int], fn: callable) -> None:
  function write_dict_to (line 29) | def write_dict_to(name: str, writer: SummaryWriter, input: dict, step: i...
  function describe (line 52) | def describe(dic, prefix="", str_built=[]) -> str:
  function preprocess (line 95) | def preprocess(dic: any, height=224, width=224):
  function format_imgs (line 127) | def format_imgs(dic: any, sz: int):

FILE: rtx/efficient_net.py
  class EfficientNetFilm (line 7) | class EfficientNetFilm(nn.Module):
    method __init__ (line 33) | def __init__(
    method __call__ (line 56) | def __call__(self, img: str):

FILE: rtx/rtx1.py
  function exists (line 20) | def exists(val):
  function default (line 24) | def default(val, d):
  function cast_tuple (line 28) | def cast_tuple(val, length=1):
  function pack_one (line 32) | def pack_one(x, pattern):
  function unpack_one (line 36) | def unpack_one(x, ps, pattern):
  function posemb_sincos_1d (line 43) | def posemb_sincos_1d(seq, dim, temperature=10000, device=None, dtype=tor...
  class Residual (line 56) | class Residual(nn.Module):
    method __init__ (line 57) | def __init__(self, fn):
    method forward (line 61) | def forward(self, x):
  class FeedForward (line 65) | class FeedForward(nn.Module):
    method __init__ (line 66) | def __init__(self, dim, mult=4, dropout=0.0):
    method forward (line 79) | def forward(self, x, cond_fn=None):
  class SqueezeExcitation (line 92) | class SqueezeExcitation(nn.Module):
    method __init__ (line 93) | def __init__(self, dim, shrinkage_rate=0.25):
    method forward (line 106) | def forward(self, x):
  class MBConvResidual (line 110) | class MBConvResidual(nn.Module):
    method __init__ (line 111) | def __init__(self, fn, dropout=0.0):
    method forward (line 116) | def forward(self, x):
  class Dropsample (line 122) | class Dropsample(nn.Module):
    method __init__ (line 123) | def __init__(self, prob=0):
    method forward (line 127) | def forward(self, x):
  function MBConv (line 140) | def MBConv(
  class Attention (line 180) | class Attention(nn.Module):
    method __init__ (line 181) | def __init__(self, dim, dim_head=32, dropout=0.0, window_size=7):
    method forward (line 215) | def forward(self, x):
  class FilmViTConfig (line 284) | class FilmViTConfig:
    method __init__ (line 287) | def __init__(
  class FilmMaxVit (line 365) | class FilmMaxVit(nn.Module):
    method __init__ (line 366) | def __init__(
    method forward (line 489) | def forward(
  class TransformerAttention (line 518) | class TransformerAttention(nn.Module):
    method __init__ (line 519) | def __init__(
    method forward (line 548) | def forward(
  class Transformer (line 604) | class Transformer(nn.Module):
    method __init__ (line 605) | def __init__(
    method forward (line 630) | def forward(
  class TokenLearner (line 654) | class TokenLearner(nn.Module):
    method __init__ (line 660) | def __init__(self, *, dim, ff_mult=2, num_output_tokens=8, num_layers=2):
    method forward (line 681) | def forward(self, x):
  class RT1Config (line 697) | class RT1Config:
    method __init__ (line 698) | def __init__(
  class RT1 (line 738) | class RT1(nn.Module):
    method __init__ (line 739) | def __init__(
    method embed_texts (line 796) | def embed_texts(self, texts: List[str]):
    method forward (line 800) | def forward(
  class RTX1 (line 884) | class RTX1(nn.Module):
    method __init__ (line 905) | def __init__(
    method train (line 955) | def train(self, video, instructions):
    method run (line 978) | def run(self, video, instructions, cond_scale=1.0):

FILE: rtx/rtx2.py
  class RTX2 (line 12) | class RTX2(torch.nn.Module):
    method __init__ (line 51) | def __init__(
    method forward (line 114) | def forward(self, img: torch.Tensor, text: torch.Tensor):

FILE: rtx2_example.py
  function run (line 5) | def run():

FILE: run.py
  function main (line 28) | def main(_):

FILE: tests/test_data_utils.py
  function test_describe (line 10) | def test_describe():
  function test_describe_empty (line 19) | def test_describe_empty():
  function test_describe_non_dict (line 24) | def test_describe_non_dict():
  function test_preprocess (line 29) | def test_preprocess():
  function test_preprocess_empty (line 39) | def test_preprocess_empty():
  function test_preprocess_non_dict (line 45) | def test_preprocess_non_dict():
  function test_preprocess_none_value (line 51) | def test_preprocess_none_value():
  function test_preprocess_image (line 57) | def test_preprocess_image():
  function test_format_imgs (line 66) | def test_format_imgs():
  function test_format_imgs_empty (line 76) | def test_format_imgs_empty():
  function test_format_imgs_non_dict (line 82) | def test_format_imgs_non_dict():
  function test_format_imgs_image (line 88) | def test_format_imgs_image():
  function test_format_imgs_tensor (line 97) | def test_format_imgs_tensor():
  function test_format_imgs_list (line 103) | def test_format_imgs_list():
  function test_format_imgs_nested_dict (line 109) | def test_format_imgs_nested_dict():
  function test_format_imgs_nested_list (line 115) | def test_format_imgs_nested_list():
  function test_format_imgs_nested_image (line 121) | def test_format_imgs_nested_image():
  function test_format_imgs_nested_tensor (line 131) | def test_format_imgs_nested_tensor():
  function test_format_imgs_nested_list (line 138) | def test_format_imgs_nested_list():

FILE: tests/test_rtx1.py
  class RTX1Test (line 6) | class RTX1Test(unittest.TestCase):
    method setUp (line 7) | def setUp(self):
    method test_default_pretrained_has_same_shape (line 34) | def test_default_pretrained_has_same_shape(self):
    method test_default_train_eval (line 40) | def test_default_train_eval(self):
    method test_pretrained_train_eval (line 50) | def test_pretrained_train_eval(self):

FILE: tests/tests.py
  function efficientnet_model (line 20) | def efficientnet_model():
  function test_efficientnet_init (line 26) | def test_efficientnet_init(efficientnet_model):
  function test_efficientnet_process_image (line 31) | def test_efficientnet_process_image(efficientnet_model):
  function test_efficientnet_image_resize (line 45) | def test_efficientnet_image_resize(efficientnet_model):
  function test_efficientnet_model_loading (line 61) | def test_efficientnet_model_loading(efficientnet_model):
  function test_efficientnet_image_transformations (line 67) | def test_efficientnet_image_transformations(efficientnet_model):
  function test_efficientnet_num_classes (line 81) | def test_efficientnet_num_classes(efficientnet_model):
  function test_efficientnet_missing_image (line 87) | def test_efficientnet_missing_image(efficientnet_model):
  function test_efficientnet_incorrect_image_format (line 93) | def test_efficientnet_incorrect_image_format(efficientnet_model):
  function test_efficientnet_model_selection (line 99) | def test_efficientnet_model_selection():
  function test_efficientnet_invalid_model_name (line 113) | def test_efficientnet_invalid_model_name():
  function test_efficientnet_invalid_num_classes (line 119) | def test_efficientnet_invalid_num_classes():
  function test_efficientnet_invalid_resize_size (line 125) | def test_efficientnet_invalid_resize_size():
  function test_efficientnet_incorrect_image_channels (line 131) | def test_efficientnet_incorrect_image_channels(efficientnet_model):
  function test_efficientnet_incorrect_image_size (line 146) | def test_efficientnet_incorrect_image_size(efficientnet_model):
  function rtx1_model (line 168) | def rtx1_model():
  function test_rtx1_initialization (line 174) | def test_rtx1_initialization(rtx1_model):
  function test_rtx1_train (line 181) | def test_rtx1_train(rtx1_model):
  function test_rtx1_eval (line 195) | def test_rtx1_eval(rtx1_model):
  function test_rtx1_train_with_invalid_inputs (line 209) | def test_rtx1_train_with_invalid_inputs(rtx1_model):
  function test_rtx1_eval_with_invalid_inputs (line 222) | def test_rtx1_eval_with_invalid_inputs(rtx1_model):
  function test_rtx1_conditional_scaling (line 235) | def test_rtx1_conditional_scaling(rtx1_model):
  function test_rtx1_model_selection (line 250) | def test_rtx1_model_selection():
  function test_rtx1_invalid_model_name (line 262) | def test_rtx1_invalid_model_name():
  function test_rtx1_negative_num_classes (line 268) | def test_rtx1_negative_num_classes():
  function test_rtx1_negative_dimension (line 274) | def test_rtx1_negative_dimension():
  function test_rtx1_negative_dim_conv_stem (line 280) | def test_rtx1_negative_dim_conv_stem():
  function test_rtx1_negative_dim_head_vit (line 286) | def test_rtx1_negative_dim_head_vit():
  function test_rtx1_negative_depth_vit (line 292) | def test_rtx1_negative_depth_vit():
  function test_rtx1_negative_window_size (line 298) | def test_rtx1_negative_window_size():
  function test_rtx1_negative_mbconv_expansion_rate (line 304) | def test_rtx1_negative_mbconv_expansion_rate():
  function test_rtx1_negative_mbconv_shrinkage_rate (line 310) | def test_rtx1_negative_mbconv_shrinkage_rate():
  function test_rtx1_negative_dropout_vit (line 316) | def test_rtx1_negative_dropout_vit():
  function test_rtx1_negative_num_actions (line 322) | def test_rtx1_negative_num_actions():
  function test_rtx1_negative_depth_rt1 (line 328) | def test_rtx1_negative_depth_rt1():
  function test_rtx1_negative_heads (line 334) | def test_rtx1_negative_heads():
  function test_rtx1_negative_dim_head_rt1 (line 340) | def test_rtx1_negative_dim_head_rt1():
  function test_rtx1_negative_cond_drop_prob (line 346) | def test_rtx1_negative_cond_drop_prob():
  function rtx2_model (line 356) | def rtx2_model():
  function test_rtx2_initialization (line 362) | def test_rtx2_initialization(rtx2_model):
  function test_rtx2_forward_pass (line 369) | def test_rtx2_forward_pass(rtx2_model):
  function test_rtx2_forward_with_invalid_inputs (line 379) | def test_rtx2_forward_with_invalid_inputs(rtx2_model):
  function test_rtx2_with_different_configs (line 387) | def test_rtx2_with_different_configs():
  function test_rtx2_negative_image_size (line 411) | def test_rtx2_negative_image_size():
  function test_rtx2_negative_patch_size (line 417) | def test_rtx2_negative_patch_size():
  function test_rtx2_negative_encoder_dim (line 423) | def test_rtx2_negative_encoder_dim():
  function test_rtx2_negative_encoder_depth (line 429) | def test_rtx2_negative_encoder_depth():
  function test_rtx2_negative_decoder_dim (line 435) | def test_rtx2_negative_decoder_dim():
  function test_rtx2_negative_decoder_depth (line 441) | def test_rtx2_negative_decoder_depth():
  function test_rtx2_negative_encoder_heads (line 447) | def test_rtx2_negative_encoder_heads():
  function test_rtx2_negative_decoder_heads (line 453) | def test_rtx2_negative_decoder_heads():