Functions — github.com/NX-AI/vision-lstm

Method__ge__

(self, other)

src/ksuit/utils/checkpoint.py:90

Method__getattr__

(self, name: str)

src/vislstm/modules/xlstm/components/util.py:74

Method__getattr__

(self, item)

src/ksuit/utils/noop_tqdm.py:14

Method__getattr__

(self, item)

src/ksuit/data/wrappers/mode_wrapper.py:74

Method__getattr__

(self, item)

src/ksuit/data/base/dataset.py:28

Method__getattr__

(self, item)

src/ksuit/data/base/subset.py:15

Method__getattr__

(self, item)

src/ksuit/data/base/wrapper.py:15

Method__getitem__

(self, key)

src/vislstm/modules/xlstm/components/util.py:47

Method__getitem__

(self, idx)

src/vislstm/datasets/cifar100.py:40

Method__getitem__

(self, idx)

src/vislstm/datasets/cifar10.py:40

Method__getitem__

(self, key)

src/ksuit/providers/summary_providers/primitive_summary_provider.py:25

Method__getitem__

(self, key)

src/ksuit/providers/summary_providers/noop_summary_provider.py:18

Method__getitem__

(self, key)

src/ksuit/providers/summary_providers/wandb_summary_provider.py:27

Method__getitem__

(self, key)

src/ksuit/providers/summary_providers/base/summary_provider_base.py:14

Method__getitem__

(self, idx)

src/ksuit/data/wrappers/mode_wrapper.py:46

Method__getitem__

(self, idx)

src/ksuit/data/base/dataset.py:25

Method__getitem__

(self, idx)

src/ksuit/datasets/multiclass_dataset.py:63

Method__getitem__

(self, idx)

src/ksuit/datasets/random_dataset.py:28

Method__hash__

(self)

src/ksuit/utils/checkpoint.py:87

Method__hash__

(self)

src/ksuit/models/poolings/base/pooling_base.py:28

Method__init__

(self, model, **kwargs)

src/vislstm/initializers/deit3_pretrained_initializer.py:12

Method__init__

(self, model, **kwargs)

src/vislstm/initializers/vim_pretrained_initializer.py:12

Method__init__

(self, model, **kwargs)

src/vislstm/initializers/deit_pretrained_initializer.py:12

Method__init__

(self, config: xLSTMBlockConfig)

src/vislstm/modules/xlstm/xlstm_block.py:60

Method__init__

(self, config: xLSTMBlockStackConfig)

src/vislstm/modules/xlstm/xlstm_block_stack.py:83

Method__init__

(self, **kwargs)

src/vislstm/modules/xlstm/utils.py:34

Method__init__

(self, config: LinearHeadwiseExpandConfig)

src/vislstm/modules/xlstm/components/linear_headwise.py:48

Method__init__

( self, ndim: int = -1, weight: bool = True, bias: bool = False, eps:

src/vislstm/modules/xlstm/components/ln.py:12

Method__init__

( self, module, parameter_name, internal_to_external: Callable[[torch.Tensor],

src/vislstm/modules/xlstm/components/util.py:35

Method__init__

(self, config: CausalConv1dConfig)

src/vislstm/modules/xlstm/components/conv.py:39

Method__init__

(self, config: FeedForwardConfig)

src/vislstm/modules/xlstm/components/feedforward.py:52

Method__init__

(self, config: xLSTMBlockConfig)

src/vislstm/modules/xlstm/blocks/xlstm_block.py:62

Method__init__

(self, config: mLSTMBlockConfig)

src/vislstm/modules/xlstm/blocks/mlstm/block.py:23

Method__init__

(self, config: mLSTMLayerConfig)

src/vislstm/modules/xlstm/blocks/mlstm/layer.py:55

Method__init__

(self, config: mLSTMCellConfig)

src/vislstm/modules/xlstm/blocks/mlstm/cell.py:25

Method__init__

(self, config: sLSTMBlockConfig)

src/vislstm/modules/xlstm/blocks/slstm/block.py:33

Method__init__

(self, config: sLSTMLayerConfig)

src/vislstm/modules/xlstm/blocks/slstm/layer.py:36

Method__init__

(self, config: sLSTMCellConfig, skip_backend_init=False)

src/vislstm/modules/xlstm/blocks/slstm/cell.py:576

Method__init__

skip device init is intended for converting models between hardware configurations / backends i.e. to store a model first and later c

src/vislstm/modules/xlstm/blocks/slstm/cell.py:654

Method__init__

(self, aggregate="flatten", **kwargs)

src/vislstm/poolings/bilateral.py:7

Method__init__

(self, num_tokens=1, aggregate="flatten", **kwargs)

src/vislstm/poolings/middle_tokens.py:5

Method__init__

(self, *args, **kwargs)

src/vislstm/sample_wrappers/segmentation_transform_wrapper.py:6

Method__init__

(self, dataset_key, **kwargs)

src/vislstm/callbacks/offline_classsubset_accuracy_callback.py:9

Method__init__

(self, resize_size=224, center_crop_size=224, interpolation="bicubic", **kwargs)

src/vislstm/callbacks/offline_imagenet_corruption_callback.py:15

Method__init__

(self, layerwise_lr_decay, mlp_is_layer=False, group_two_blocks=False)

src/vislstm/optim/layerwise_lr_decay.py:5

Method__init__

( self, mixup_alpha: float = None, cutmix_alpha: float = None,

src/vislstm/collators/mix_collator.py:21

Method__init__

( self, size, scale=(0.08, 1.0), ratio=(3.0 / 4.0, 4.0 / 3.0),

src/vislstm/transforms/random_resized_crop.py:14

Method__init__

(self, blur_sigma=(0.1, 2.0), **kwargs)

src/vislstm/transforms/three_augment.py:9

Method__init__

(self, *args, interpolation="bilinear", **kwargs)

src/vislstm/transforms/resize.py:10

Method__init__

(self, brightness=0, contrast=0, saturation=0, hue=0, **kwargs)

src/vislstm/transforms/color_jitter.py:8

Method__init__

(self, size, max_category_ratio=1., ignore_index=-1, **kwargs)

src/vislstm/transforms/segmentation_random_crop.py:8

Method__init__

(self, p=0.5, **kwargs)

src/vislstm/transforms/segmentation_random_horizontal_flip.py:7

Method__init__

(self, size, fill=-1, **kwargs)

src/vislstm/transforms/segmentation_pad.py:8

Method__init__

(self, **kwargs)

src/vislstm/transforms/cifar10_norm.py:5

Method__init__

(self, ratio_resolution, ratio_range, interpolation="bilinear", **kwargs)

src/vislstm/transforms/segmentation_random_resize.py:12

Method__init__

(self, **kwargs)

src/vislstm/transforms/imagenet1k_norm.py:5

Method__init__

( self, num_ops: int, magnitude: int, fill_color,

src/vislstm/transforms/rand_augment.py:30

Method__init__

(self, sigma, **kwargs)

src/vislstm/transforms/gaussian_blur_pil.py:10

Method__init__

(self, *args, interpolation="bilinear", **kwargs)

src/vislstm/transforms/segmentation_resize.py:9

Method__init__

(self, **kwargs)

src/vislstm/transforms/cifar100_norm.py:5

Method__init__

( self, global_root=None, local_root=None, dataset_config_prov

src/vislstm/datasets/imagenet_corruption.py:6

Method__init__

( self, split, global_root=None, local_root=None,

src/vislstm/datasets/cifar100.py:6

Method__init__

( self, global_root=None, local_root=None, dataset_config_prov

src/vislstm/datasets/imagenet_rendition.py:18

Method__init__

( self, global_root=None, local_root=None, dataset_config_prov

src/vislstm/datasets/imagenet_adversarial.py:18

Method__init__

( self, global_root=None, local_root=None, dataset_config_prov

src/vislstm/datasets/imagenet_sketch.py:6

Method__init__

( self, split, global_root=None, local_root=None,

src/vislstm/datasets/cifar10.py:6

Method__init__

( self, split, global_root=None, local_root=None,

src/vislstm/datasets/ade20k.py:55

Method__init__

( self, split, global_root=None, local_root=None,

src/vislstm/datasets/imagenet1k.py:6

Method__init__

(self, input_dim, **kwargs)

src/vislstm/models/upernet/upernet_postprocessor.py:10

Method__init__

( self, encoder, extractor, postprocessor, decoder

src/vislstm/models/upernet/upernet_model.py:7

Method__init__

( self, input_dim: int, dim: int = 256, kernel_size: int = 3,

src/vislstm/models/upernet/upernet_auxiliary.py:11

Method__init__

(self, input_dim, dim=512, dropout=0.0, **kwargs)

src/vislstm/models/upernet/upernet_decoder.py:11

Method__init__

( self, patch_size, dim, depth, num_attn_heads,

src/vislstm/models/single/vit.py:22

Method__init__

( self, patch_size, dim, depth, stride=None,

src/vislstm/models/single/vislstm.py:20

Method__init__

( self, patch_size, dim, depth, drop_path_rate=0.0

src/vislstm/models/single/vismamba.py:14

Method__init__

(self, img_size=224, patch_size=16, stride=16, in_chans=3, embed_dim=768, norm_layer=None, fl

src/vislstm/external/models_mamba.py:233

Method__init__

(self, img_size=224, patch_size=16, stride=16,

src/vislstm/external/models_mamba.py:420

Method__init__

(self, classifier_path, init="mae", **kwargs)

src/ksuit/initializers/reset_classifier_head_initializer.py:14

Method__init__

(self, repo, model, source="github", missing_keys=None, unexpected_keys=None, **kwargs)

src/ksuit/initializers/torchhub_initializer.py:12

Method__init__

( self, filename, base_path=None, weights_key=None,

src/ksuit/initializers/state_dict_initializer.py:11

Method__init__

(self, load_optim=True, **kwargs)

src/ksuit/initializers/resume_initializer.py:15

Method__init__

( self, load_optim=False, keys_to_remove=None, patterns_to_rem

src/ksuit/initializers/previous_run_initializer.py:10

Method__init__

(self, path_provider: PathProvider = None)

src/ksuit/initializers/base/initializer_base.py:7

Method__init__

( self, stage_id, model_name, checkpoint, load_opt

src/ksuit/initializers/base/checkpoint_initializer.py:9

Method__init__

(self, config: dict)

src/ksuit/configs/static_config.py:21

Method__init__

(self, mode: str, host: str = None, entity: str = None, project: str = None)

src/ksuit/configs/wandb_config.py:4

Method__init__

(self, effective_batch_size, updates_per_epoch)

src/ksuit/configs/resolver_processors/mindata_postprocessor.py:10

Method__init__

(self, effective_batch_size, max_epochs, max_updates, max_samples)

src/ksuit/configs/resolver_processors/minduration_postprocessor.py:7

Method__init__

( self, exclude_patterns: list[str] = None, pattern_matcher: PatternMatche

src/ksuit/freezers/full_freezer.py:7

Method__init__

( self, patterns: list[str], pattern_matcher: PatternMatcher = None,

src/ksuit/freezers/pattern_freezer.py:7

Method__init__

(self, update_counter: UpdateCounter = None, schedule=None, set_to_none=False)

src/ksuit/freezers/base/freezer_base.py:11

Method__init__

( self, dataset_wrapper_factory: SingleFactory = None, sample_wrapper_fact

src/ksuit/factory/dataset_factory.py:7

Method__init__

(self)

src/ksuit/factory/base/factory_base.py:6

Method__init__

(self, key)

src/ksuit/factory/base/empty_factory.py:5

Method__init__

(self, module_names=None, kind_to_ctor=None, **kwargs)

src/ksuit/factory/base/single_factory.py:10

Method__init__

(self, seed)

src/ksuit/utils/seed.py:31

Method__init__

( self, start_checkpoint: Checkpoint, end_checkpoint: Checkpoint,

src/ksuit/utils/update_counter.py:5

Method__init__

( self, num_workers=None, max_num_workers=None, pin_memory=Non

src/ksuit/utils/data_container.py:18

Method__init__

(self, epoch=None, update=None, sample=None)

src/ksuit/utils/checkpoint.py:17

Method__init__

(self, schedule: ScheduleBase, update_counter: UpdateCounter = None)

src/ksuit/utils/schedule_wrapper.py:7

Functions1,346 in github.com/NX-AI/vision-lstm