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

↓ 2 callersMethodupdate

( self, batch, ddp_model, model=None, training=Tru

src/ksuit/trainers/base/sgd_trainer.py:810

↓ 2 callersFunctionwang_init_

Adopted from https://github.com/EleutherAI/gpt-neox/blob/main/megatron/model/init_functions.py.

src/vislstm/modules/xlstm/components/init.py:29

↓ 2 callersFunctionwang_init_

Adopted from https://github.com/EleutherAI/gpt-neox/blob/main/megatron/model/init_functions.py.

vision_lstm/vision_lstm2.py:42

↓ 1 callersMethod__init__

(self, **kwargs)

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

↓ 1 callersMethod__init__

(self, loss_kind="cross_entropy", forward_kwargs=None, **kwargs)

src/ksuit/trainers/classification_trainer.py:11

↓ 1 callersMethod__init__

(self, loss_weights=None, ignore_index=-1, **kwargs)

src/ksuit/trainers/segmentation_trainer.py:11

↓ 1 callersMethod__post_init__

(self)

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

↓ 1 callersMethod_after_initializers

(self)

src/ksuit/models/base/single_model.py:136

↓ 1 callersMethod_after_initializers

(self)

src/ksuit/models/base/composite_model.py:101

↓ 1 callersMethod_after_training

(self, **kwargs)

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

↓ 1 callersMethod_automatic_max_batch_size

( self, effective_batch_size_per_device, model, ddp_model,

src/ksuit/trainers/base/sgd_trainer.py:279

↓ 1 callersMethod_before_training

(self, **kwargs)

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

↓ 1 callersMethod_bias_ext2int

(self, bias_ext: torch.Tensor)

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

↓ 1 callersMethod_calculate_batch_size_and_accumulation_steps

(self, model, ddp_model)

src/ksuit/trainers/base/sgd_trainer.py:366

↓ 1 callersMethod_check_input

(self, input: torch.Tensor)

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

↓ 1 callersMethod_collate_result

(result, global_dataset_len)

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

↓ 1 callersMethod_collate_tensors

(tensors)

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

↓ 1 callersMethod_create_block_map

Creates the block map, that specifies which block is used at which position.

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

↓ 1 callersMethod_create_blocks

(self, config: xLSTMBlockStackConfig)

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

↓ 1 callersMethod_create_weight_decay_optim_groups

Return a tuple of two sequences, one for parameters with weight decay and one for parameters without weight decay. Default separation:

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

↓ 1 callersFunction_get_device_count

()

src/ksuit/utils/num_worker_heuristic.py:34

↓ 1 callersMethod_get_final_state

All states has the structure [STATES, SEQUENCE, BATCH, HIDDEN]

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

↓ 1 callersMethod_get_model_kwargs

(self)

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

↓ 1 callersMethod_get_model_state_dict

(self, model)

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

↓ 1 callersMethod_get_ops

(self)

src/vislstm/transforms/rand_augment.py:65

↓ 1 callersMethod_get_own_outputs

(self)

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

↓ 1 callersMethod_get_param_counts

(model, trace=None)

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

↓ 1 callersMethod_get_state

( self, input: torch.Tensor, state: Optional[torch.Tensor] = None )

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

↓ 1 callersMethod_has_param_with_grad

(self)

src/ksuit/optim/optim_wrapper.py:195

↓ 1 callersMethod_impl

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

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

↓ 1 callersMethod_init_optim

(self, name, model)

src/ksuit/initializers/resume_initializer.py:36

↓ 1 callersMethod_init_weights

(self, name, model)

src/ksuit/initializers/resume_initializer.py:21

↓ 1 callersMethod_is_new_best_model

(self, metric_value)

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

↓ 1 callersFunction_lists_to_dict

wandb cant handle lists in configs -> transform lists into dicts with str(i) as key

src/ksuit/utils/wandb_utils.py:127

↓ 1 callersFunction_lists_to_dicts_impl

(root)

src/ksuit/utils/wandb_utils.py:134

↓ 1 callersMethod_metric_improved

(self, cur_metric)

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

↓ 1 callersFunction_ntuple

(n)

vision_lstm/vision_lstm_util.py:13

↓ 1 callersMethod_permute_input

# TODO Adapt this # >>> BaseRNN(BaseRNNConfig(10, 10, num_heads=2, input_shape='SBG'))._permute_input(torch.zeros((5, 2, 10))).shape

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

↓ 1 callersMethod_permute_output

>>> BaseRNN(BaseRNNConfig(10, 16, num_heads=2, output_shape='SBH'))._permute_output(torch.zeros((5, 2, 16))).shape torch.Size([5, 2,

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

↓ 1 callersMethod_prepare_batch_size

(self, model, ddp_model)

src/ksuit/trainers/base/sgd_trainer.py:932

↓ 1 callersMethod_prepare_model

(self, model)

src/ksuit/trainers/base/sgd_trainer.py:471

↓ 1 callersFunction_preprocess

( train_x, test_x, train_y, test_y, k, tau, batch_norm

src/ksuit/metrics/multiclass_knn.py:16

↓ 1 callersMethod_recurrent_kernel_ext2int

( self, recurrent_kernel_ext: torch.Tensor )

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

↓ 1 callersMethod_register_hooks

(self, model)

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

↓ 1 callersMethod_register_sampler_configs

(self, trainer)

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

↓ 1 callersFunction_run_managed_multiprocess

(accelerator, main)

src/ksuit/distributed/run/managed.py:57

↓ 1 callersFunction_run_managed_singleprocess

(accelerator, main)

src/ksuit/distributed/run/managed.py:50

↓ 1 callersMethod_sample_transforms

(self)

src/vislstm/transforms/rand_augment.py:87

↓ 1 callersMethod_save_seperate_models

( self, name, model, ckpt, save_weights,

src/ksuit/callbacks/base/writers/checkpoint_writer.py:67

↓ 1 callersMethod_set_to_none

(self, model)

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

↓ 1 callersFunction_setup_master_factory

()

src/ksuit/factory/__init__.py:9

↓ 1 callersMethod_should_include_key

(key)

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

↓ 1 callersMethod_should_include_key

(key)

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

↓ 1 callersMethod_to_string

(self)

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

↓ 1 callersMethod_track_after_accumulation_step

(self, **kwargs)

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

↓ 1 callersMethod_track_after_update_step

(self, **kwargs)

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

↓ 1 callersMethod_train

( self, model, trainer_model, ddp_model, batch_siz

src/ksuit/trainers/base/sgd_trainer.py:603

↓ 1 callersMethod_wrap_model

(model)

src/ksuit/trainers/base/sgd_trainer.py:532

↓ 1 callersMethod_zero_state

Returns a zeros state matching dtype and batch size of `input`. Arguments: input: Tensor, to specify the device and dtype of the re

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

↓ 1 callersMethodadd_base_path

(base_path)

src/ksuit/factory/base/master_factory.py:12

↓ 1 callersMethodadd_samples

(self, num_samples)

src/ksuit/utils/update_counter.py:70

↓ 1 callersMethodafter_initializers

(self)

src/ksuit/models/base/model_base.py:103

↓ 1 callersMethodafter_training

(self, **kwargs)

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

↓ 1 callersMethodafter_weight_init

(self, model)

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

↓ 1 callersMethodallocate_inference_cache

(self, batch_size, max_seqlen, dtype=None, **kwargs)

src/vislstm/external/models_mamba.py:329

↓ 1 callersMethodapply_initializers

(self)

src/ksuit/models/base/model_base.py:100

↓ 1 callersMethodapply_resume_initializer

(self, model)

src/ksuit/trainers/base/sgd_trainer.py:477

↓ 1 callersMethodbackward

(ctx, *grads)

src/ksuit/distributed/functional/all_gather_grad_autograd.py:22

↓ 1 callersMethodbefore_accumulation_step

before_accumulation_step hook (e.g. for freezers)

src/ksuit/models/base/model_base.py:126

↓ 1 callersMethodbefore_every_accumulation_step

(self, **kwargs)

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

↓ 1 callersMethodbefore_every_backward

(self, **kwargs)

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

↓ 1 callersMethodbefore_every_update

(self, **kwargs)

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

↓ 1 callersMethodbefore_optim_step

(self, model)

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

↓ 1 callersFunctionbias_linspace_init_

Linearly spaced bias init across dimensions.

vision_lstm/vision_lstm.py:21

↓ 1 callersFunctionbias_linspace_init_

Linearly spaced bias init across dimensions.

vision_lstm/vision_lstm2.py:21

↓ 1 callersMethodcall_after_training

(self, trainer_model, callbacks)

src/ksuit/trainers/base/sgd_trainer.py:974

↓ 1 callersMethodcall_before_training

(self, trainer_model, batch_size, callbacks)

src/ksuit/trainers/base/sgd_trainer.py:958

↓ 1 callersFunctioncheck_all_none

(*args)

src/ksuit/utils/param_checking.py:71

↓ 1 callersFunctioncheck_at_most_one

(*args)

src/ksuit/utils/param_checking.py:67

↓ 1 callersFunctioncheck_exclusive

(*args)

src/ksuit/utils/param_checking.py:55

↓ 1 callersMethodclassify

(self, x)

src/ksuit/models/linear_model.py:17

↓ 1 callersMethodcollator

(self)

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

↓ 1 callersFunctioncreate_block

( d_model, ssm_cfg=None, norm_epsilon=1e-5, drop_path=0., rms_norm=Fal

src/vislstm/external/models_mamba.py:333

↓ 1 callersFunctioncreate_block_stack

( dim, depth, context_length, bidirectional=False, quaddirectional=Fal

src/vislstm/modules/xlstm/__init__.py:6

↓ 1 callersFunctionctor_from_name

(module_names, type_name, **kwargs)

src/ksuit/factory/base/reflection.py:68

↓ 1 callersMethoddevice

(self)

src/ksuit/models/base/composite_model.py:68

↓ 1 callersFunctiondict_to_string

{epoch: 5, batch_size: 64} --> epoch=5-batchsize=64

src/ksuit/utils/formatting_utils.py:59

↓ 1 callersFunctionfinish_wandb

(wandb_config: WandbConfig)

src/ksuit/utils/wandb_utils.py:145

↓ 1 callersFunctionflatten_transform

(transform)

src/ksuit/utils/transform_utils.py:9

↓ 1 callersFunctionfname_to_loader

(fname: str)

src/ksuit/data/loaders/__init__.py:23

↓ 1 callersFunctionfolder_contains_mostly_zips

(path)

src/ksuit/data/copy/utils.py:83

↓ 1 callersMethodforward

(self, input, state=None, lengths=None)

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

↓ 1 callersMethodforward

(self, x)

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

↓ 1 callersMethodforward

(self, x)

src/ksuit/models/linear_model.py:14

↓ 1 callersMethodforward

(self, x, residual_path=None, residual_path_kwargs=None)

vision_lstm/vision_lstm_util.py:197

↓ 1 callersMethodforward_features

(self, x, inference_params=None, if_random_cls_token_position=False, if_random_token_

src/vislstm/external/models_mamba.py:571

↓ 1 callersMethodfrom_zip

(src, dst)

src/vislstm/datasets/ade20k.py:37

↓ 1 callersFunctiongeneric_global_to_local

( global_root, local_root, copy_global_to_local_fn, copy_global_to_local_kwarg

src/ksuit/data/copy/utils.py:13

↓ 1 callersFunctionget_act_fn

(act_fn_name: str)

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

↓ 1 callersMethodget_all_callbacks

(self, model=None)

src/ksuit/trainers/base/sgd_trainer.py:203

← previous next →201–300 of 1,346, ranked by callers