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

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

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

⨍ Functions 1,346◇ Types & classes 277

Method_before_training

(self, model, **kwargs)

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

Method_before_training

(self, **kwargs)

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

Method_before_training

(self, **_)

src/ksuit/callbacks/default_callbacks/dataset_stats_callback.py:5

Method_before_training

(self, model, **_)

src/ksuit/callbacks/default_callbacks/param_count_callback.py:37

Method_before_training

(self, model, **_)

src/ksuit/callbacks/default_callbacks/copy_previous_summary_callback.py:14

Method_before_training

(self, model, **_)

src/ksuit/callbacks/default_callbacks/copy_previous_config_callback.py:20

Method_before_training

(self, **kwargs)

src/ksuit/callbacks/default_callbacks/progress_callback.py:14

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/online_callbacks/online_accuracy_callback.py:18

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/online_callbacks/online_segmentation_callback.py:18

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/monitor_callbacks/nan_hook_monitor_callback.py:11

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/monitor_callbacks/gradient_spike_monitor_callback.py:13

Method_before_training

(self, trainer_model, **kwargs)

src/ksuit/callbacks/monitor_callbacks/nan_loss_monitor_callback.py:17

Method_before_training

(self, model, **kwargs)

src/ksuit/callbacks/monitor_callbacks/activation_monitor_callback.py:11

Method_bias

(self)

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

Method_bias_ext2int

(self, bias_ext: torch.Tensor)

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

Method_bias_ext2int

(self, bias_ext: torch.Tensor)

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

Method_bias_int2ext

(self, bias_int: torch.Tensor)

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

Method_bias_int2ext

>>> (); mod = sLSTMCell_vanilla(sLSTMCellConfig(hidden_size=64, num_heads=2), skip_backend_init=True); () # doctest:+ELLIPSIS (...)

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

Method_bias_int2ext

>>> (); mod = sLSTMCell_cuda( ... sLSTMCellConfig(hidden_size=64, num_heads=2), skip_backend_init=True); () # doctest:+ELLIPSIS

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

Method_call_getall

(self, item)

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

Method_call_getitem

(self, func, idx, ctx=None, *args, **kwargs)

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

Method_call_getitem

(self, func, idx, ctx=None, *args, **kwargs)

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

Method_check

(key)

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

Function_check_global_root_exists

(global_root)

src/ksuit/data/copy/multiclass_dataset.py:10

Function_copy_global_to_local

(global_root, local_root, num_workers, log_fn)

src/ksuit/data/copy/multiclass_dataset.py:18

Method_create_weight_decay_optim_groups

(self)

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

Function_devices

(devices)

src/ksuit/configs/cli_args.py:92

Method_expected_x_ndim

(self)

vision_lstm/vision_lstm_util.py:137

Method_forward

(self, batch, model, trainer)

src/vislstm/callbacks/offline_classsubset_accuracy_callback.py:24

Method_forward

(batch, model, trainer)

src/vislstm/callbacks/offline_imagenet_corruption_callback.py:69

Method_forward

(self, batch, trainer_model, trainer)

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

Method_forward

(self, batch, model, trainer)

src/ksuit/callbacks/offline_callbacks/offline_segmentation_callback.py:24

Method_forward

(self, batch, model, trainer)

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

Method_forward

(self, batch, trainer_model, trainer)

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

Method_forward

(self, batch, trainer_model, trainer)

src/ksuit/callbacks/offline_callbacks/offline_loss_callback.py:17

Method_forward

(self, batch, model, trainer)

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

Method_forward_path

(self, x)

vision_lstm/vision_lstm.py:493

Method_forward_path

(self, x)

vision_lstm/vision_lstm2.py:546

Method_get_model_kwargs

(self)

src/vislstm/initializers/deit3_pretrained_initializer.py:16

Method_get_model_kwargs

(self)

src/vislstm/initializers/vim_pretrained_initializer.py:16

Method_get_model_kwargs

(self)

src/vislstm/initializers/deit_pretrained_initializer.py:16

Method_get_model_kwargs

(self)

src/ksuit/initializers/state_dict_initializer.py:35

Method_get_model_kwargs

(self)

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

Method_get_weight_decay_optim_groups_for_modules

( self, modules: list["WeightDecayOptimGroupMixin"], **kwargs )

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

Method_getitem_from_ctx

(_, ctx, ctx_key)

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

Method_getitem_index

(idx)

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

Function_hp

(hp)

src/ksuit/configs/cli_args.py:85

Method_impl

( self, training: bool, input: torch.Tensor, state: torch.Tensor )

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

Method_init_qkv_proj

(qkv_proj: LinearHeadwiseExpand)

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

Method_init_qkv_proj

(qkv_proj: LinearHeadwiseExpand)

vision_lstm/vision_lstm.py:466

Method_init_qkv_proj

(qkv_proj: LinearHeadwiseExpand)

vision_lstm/vision_lstm2.py:495

Function_init_weights

( module, n_layer, initializer_range=0.02, # Now only used for embedding layer.

src/vislstm/external/models_mamba.py:372

Method_init_weights

(module)

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

Method_init_weights

(module)

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

Method_is_cuda

(self)

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

Method_is_in_sequence

(param: nn.Parameter, sequence: Sequence[nn.Parameter])

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

Method_periodic_callback

(self, model, trainer, batch_size, data_iter, **_)

src/vislstm/callbacks/offline_classsubset_accuracy_callback.py:38

Method_periodic_callback

(self, model, trainer, batch_size, data_iter, **_)

src/vislstm/callbacks/offline_imagenet_corruption_callback.py:79

Method_periodic_callback

(self, model, trainer_model, trainer, batch_size, data_iter, **_)

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

Method_periodic_callback

(self, model, trainer, batch_size, data_iter, **_)

src/ksuit/callbacks/offline_callbacks/offline_segmentation_callback.py:137

Method_periodic_callback

(self, model, trainer, batch_size, data_iter, **_)

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

Method_periodic_callback

(self, model, trainer_model, trainer, batch_size, data_iter, **_)

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

Method_periodic_callback

(self, model, trainer_model, trainer, batch_size, data_iter, **_)

src/ksuit/callbacks/offline_callbacks/offline_loss_callback.py:35

Method_periodic_callback

(self, model, trainer, batch_size, data_iter, **_)

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

Method_periodic_callback

(self, trainer, model, **kwargs)

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

Method_periodic_callback

(self, model, trainer, **kwargs)

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

Method_periodic_callback

(self, model, **kwargs)

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

Method_periodic_callback

(self, interval_type, **_)

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

Method_periodic_callback

(self, model, **_)

src/ksuit/callbacks/default_callbacks/lr_callback.py:13

Method_periodic_callback

(self, trainer, **_)

src/ksuit/callbacks/default_callbacks/online_loss_callback.py:19

Method_periodic_callback

(self, **_)

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

Method_periodic_callback

(self, model, **__)

src/ksuit/callbacks/default_callbacks/peak_memory_callback.py:10

Method_periodic_callback

(self, trainer, interval_type, **_)

src/ksuit/callbacks/default_callbacks/progress_callback.py:18

Method_periodic_callback

(self, model, **_)

src/ksuit/callbacks/default_callbacks/freezer_callback.py:13

Method_periodic_callback

(self, **_)

src/ksuit/callbacks/online_callbacks/update_output_callback.py:60

Method_periodic_callback

(self, **_)

src/ksuit/callbacks/online_callbacks/online_accuracy_callback.py:39

Method_periodic_callback

(self, **_)

src/ksuit/callbacks/online_callbacks/online_segmentation_callback.py:38

Method_periodic_callback

(self, model, **kwargs)

src/ksuit/callbacks/monitor_callbacks/kill_on_loss_spike_callback.py:27

Method_recurrent_kernel

(self)

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

Method_recurrent_kernel_ext2int

( self, recurrent_kernel_ext: torch.Tensor )

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

Method_recurrent_kernel_ext2int

( self, recurrent_kernel_ext: torch.Tensor )

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

Method_recurrent_kernel_int2ext

( self, recurrent_kernel_int: torch.Tensor )

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

Method_recurrent_kernel_int2ext

>>> (); mod = sLSTMCell_vanilla(sLSTMCellConfig(hidden_size=64, num_heads=2), skip_backend_init=True); () # doctest:+ELLIPSIS (...)

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

Method_recurrent_kernel_int2ext

>>> (); mod = sLSTMCell_cuda( ... sLSTMCellConfig(hidden_size=64, num_heads=2), skip_backend_init=True); () # doctest:+ELLIPSIS

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

Method_register_hooks

(self, model)

src/ksuit/models/extractors/block_extractor.py:37

Method_register_hooks

(self, model)

src/ksuit/models/extractors/single_extractor.py:5

Method_register_sampler_config_from_dataset

(self, dataset, mode, name)

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

Method_register_sampler_configs

(self, trainer)

src/vislstm/callbacks/offline_classsubset_accuracy_callback.py:21

Method_register_sampler_configs

(self, trainer)

src/vislstm/callbacks/offline_imagenet_corruption_callback.py:64

Method_register_sampler_configs

(self, trainer)

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

Method_register_sampler_configs

(self, trainer)

src/ksuit/callbacks/offline_callbacks/offline_segmentation_callback.py:21

Method_register_sampler_configs

(self, trainer)

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

Method_register_sampler_configs

(self, trainer)

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

Method_register_sampler_configs

(self, trainer)

src/ksuit/callbacks/offline_callbacks/offline_loss_callback.py:14

Method_register_sampler_configs

(self, trainer)

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

Function_run_multiprocess

(rank, accelerator, device_ids, master_port, world_size, main)

src/ksuit/distributed/run/unmanaged.py:40

Method_should_stop

(self)

src/ksuit/trainers/early_stoppers/metric_early_stopper.py:21

Method_to_string

(self)

src/ksuit/callbacks/online_callbacks/update_output_callback.py:39

Method_track_after_accumulation_step

(self, losses, **kwargs)

src/ksuit/callbacks/default_callbacks/online_loss_callback.py:15

Method_track_after_accumulation_step

(self, update_outputs, **kwargs)

src/ksuit/callbacks/online_callbacks/update_output_callback.py:42

← previous next →801–900 of 1,346, ranked by callers