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

CodeHub MCPcopy Create free account

hub / github.com/NX-AI/vision-lstm / functions

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

⨍ Functions 1,346◇ Types & classes 277

(self, seqlens, dim: int, allow_interpolation: bool = True)

vision_lstm/vision_lstm_util.py:128

(self, dim, num_heads, bias=False)

vision_lstm/vision_lstm2.py:140

(self, dim, kernel_size=4, bias=True)

vision_lstm/vision_lstm2.py:194

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

vision_lstm/vision_lstm2.py:228

(self, dim, num_heads, norm_bias=True)

vision_lstm/vision_lstm2.py:292

( self, dim, direction, drop_path=0.0, conv_kind="

vision_lstm/vision_lstm2.py:509

( self, dim, drop_path=0.0, conv_kind="2d", conv_k

vision_lstm/vision_lstm2.py:561

( self, dim=192, input_shape=(3, 224, 224), patch_size=16,

vision_lstm/vision_lstm2.py:606

( self, dim, depth, num_heads, patch_size=16,

vision_lstm/vision_transformer.py:9

src/ksuit/utils/checkpoint.py:195

src/ksuit/utils/noop_tqdm.py:17

torch.utils.data.Dataset doesn't define __iter__ which makes 'for sample in dataset' run endlessly

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

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

src/vislstm/datasets/cifar100.py:43

src/vislstm/datasets/cifar10.py:43

src/vislstm/datasets/ade20k.py:121

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

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

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

src/ksuit/datasets/multiclass_dataset.py:66

src/ksuit/datasets/random_dataset.py:31

(cls, config: sLSTMCellConfig, skip_backend_init: bool = False)

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

Method__post_init__

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

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

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

src/ksuit/utils/checkpoint.py:106

src/ksuit/callbacks/base/callback_base.py:70

src/ksuit/optim/param_group_modifiers/base/param_group_modifier_base.py:5

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

src/ksuit/models/extractors/finalizers/mean_finalizer.py:8

src/ksuit/models/extractors/finalizers/concat_finalizer.py:11

src/ksuit/models/extractors/finalizers/index_finalizer.py:8

src/ksuit/models/extractors/finalizers/stack_finalizer.py:11

src/ksuit/models/extractors/base/extractor_base.py:36

Method__setitem__

(self, key, value)

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

Method__setitem__

(self, key, value)

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

Method__setitem__

(self, key, value)

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

Method__setitem__

(self, key, value)

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

Method__setitem__

(self, key, value)

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

Method__setitem__

(self, key, value)

src/ksuit/providers/config_providers/noop_config_provider.py:8

Method__setitem__

(self, key, value)

src/ksuit/providers/config_providers/primitive_config_provider.py:17

Method__setitem__

(self, key, value)

src/ksuit/providers/config_providers/wandb_config_provider.py:17

Method__setitem__

(self, key, value)

src/ksuit/providers/config_providers/base/config_provider_base.py:5

src/vislstm/poolings/first_token.py:12

src/vislstm/poolings/bilateral.py:27

src/vislstm/poolings/last_token.py:12

src/vislstm/poolings/middle_tokens.py:29

src/vislstm/optim/layerwise_lr_decay.py:82

src/vislstm/datasets/cifar100.py:27

src/vislstm/datasets/cifar10.py:27

src/ksuit/freezers/full_freezer.py:17

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

src/ksuit/utils/checkpoint.py:109

src/ksuit/callbacks/base/callback_base.py:73

src/ksuit/callbacks/base/periodic_callback.py:60

src/ksuit/optim/lr_scalers/linear_lr_scaler.py:9

src/ksuit/optim/lr_scalers/sqrt_lr_scaler.py:11

src/ksuit/optim/lr_scalers/lr_scaler_base.py:2

src/ksuit/optim/param_group_modifiers/lr_scale_by_name_modifier.py:18

src/ksuit/optim/param_group_modifiers/weight_decay_by_name_modifier.py:26

src/ksuit/optim/param_group_modifiers/base/param_group_modifier_base.py:8

src/ksuit/models/poolings/class_token.py:29

src/ksuit/models/poolings/max_patch.py:12

src/ksuit/models/poolings/mean_patch.py:18

src/ksuit/models/poolings/concat_class_average.py:17

src/ksuit/models/poolings/identity.py:11

src/ksuit/models/poolings/mean_all.py:12

src/ksuit/models/poolings/max_all.py:12

src/ksuit/models/poolings/to_image.py:47

src/ksuit/models/poolings/all_patches.py:14

src/ksuit/models/poolings/index_pooling.py:16

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

src/ksuit/models/extractors/finalizers/mean_finalizer.py:11

src/ksuit/models/extractors/finalizers/concat_finalizer.py:14

src/ksuit/models/extractors/finalizers/index_finalizer.py:11

src/ksuit/models/extractors/finalizers/stack_finalizer.py:14

src/ksuit/models/extractors/base/extractor_base.py:39

src/ksuit/utils/checkpoint.py:188

Method_after_training

(self, **kwargs)

src/ksuit/callbacks/checkpoint_callbacks/best_checkpoint_callback.py:96

Method_after_training

(self, model, trainer, **kwargs)

src/ksuit/callbacks/checkpoint_callbacks/checkpoint_callback.py:68

Method_after_training

(self, model, **kwargs)

src/ksuit/callbacks/checkpoint_callbacks/ema_callback.py:113

Method_after_training

src/ksuit/callbacks/default_callbacks/train_time_callback.py:43

Method_after_training

src/ksuit/callbacks/default_callbacks/eta_callback.py:135

Method_before_training

(self, model, **kwargs)

src/vislstm/callbacks/offline_classsubset_accuracy_callback.py:16

Method_before_training

(self, model, **kwargs)

src/vislstm/callbacks/offline_imagenet_corruption_callback.py:43

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/offline_callbacks/offline_multiclass_knn_callback.py:44

Method_before_training

(self, model, trainer, **kwargs)

src/ksuit/callbacks/offline_callbacks/offline_accuracy_callback.py:19

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/offline_callbacks/offline_features_callback.py:28

Method_before_training

(self, **kwargs)

src/ksuit/callbacks/visualization_callbacks/visualize_segmentation_callback.py:29

Method_before_training

(self, **kwargs)

src/ksuit/callbacks/checkpoint_callbacks/best_checkpoint_callback.py:45

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/checkpoint_callbacks/checkpoint_callback.py:26

← previous next →701–800 of 1,346, ranked by callers