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

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm.py:211

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm.py:262

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm.py:342

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm.py:502

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm_util.py:107

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm_util.py:140

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm2.py:154

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm2.py:211

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm2.py:262

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm2.py:342

↓ 1 callersMethodreset_parameters

vision_lstm/vision_lstm2.py:555

↓ 1 callersMethodresume_from_checkpoint

(self, stage_name, stage_id, model)

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

↓ 1 callersMethodrun

src/ksuit/runners/runner.py:38

↓ 1 callersFunctionrun_unzip_jobs

(jobargs, num_workers)

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

↓ 1 callersFunctionsLSTMCellFuncGenerator

(training, config: sLSTMCellConfig)

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

↓ 1 callersMethodsample

src/ksuit/utils/update_counter.py:57

↓ 1 callersMethodsave_resolved_hp

(stage_hp, out_file_uri)

src/ksuit/configs/hyperparams.py:39

↓ 1 callersMethodsave_unresolved_hp

(hp_file, out_file_uri)

src/ksuit/configs/hyperparams.py:33

↓ 1 callersMethodscale

(self, factor, updates_per_epoch, effective_batch_size, floor)

src/ksuit/utils/checkpoint.py:56

↓ 1 callersMethodscale_lr

(self, base_lr, lr_scale_factor)

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

↓ 1 callersMethodschedule_step

src/ksuit/optim/optim_wrapper.py:226

↓ 1 callersMethodset_accumulation_steps

(self, accumulation_steps)

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

↓ 1 callersMethodset_backward

(self, key, value)

src/ksuit/utils/bidict.py:29

↓ 1 callersMethodset_rng

src/ksuit/data/collators/stochastic_collator.py:40

↓ 1 callersMethodset_rng

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

↓ 1 callersMethodset_rng

src/ksuit/data/transforms/base/stochastic_transform.py:39

↓ 1 callersMethodshould_log_after_update

(self, checkpoint)

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

↓ 1 callersMethodshould_log_after_update

(self, checkpoint)

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

↓ 1 callersMethodshould_stop_after_epoch

(self, checkpoint)

src/ksuit/trainers/early_stoppers/base/early_stopper_base.py:44

↓ 1 callersMethodshould_stop_after_sample

(self, checkpoint, effective_batch_size)

src/ksuit/trainers/early_stoppers/base/early_stopper_base.py:30

↓ 1 callersMethodshould_stop_after_update

(self, checkpoint)

src/ksuit/trainers/early_stoppers/base/early_stopper_base.py:39

↓ 1 callersMethodsolarize

src/vislstm/transforms/rand_augment.py:125

↓ 1 callersMethodstate_dict

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

↓ 1 callersMethodsummarize_logvalues

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

↓ 1 callersMethodto_forward

src/ksuit/utils/bidict.py:13

↓ 1 callersFunctionto_one_hot_vector

src/ksuit/utils/one_hot_utils.py:5

↓ 1 callersMethodto_string

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

↓ 1 callersMethodto_string

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

↓ 1 callersFunctiontotal_area_to_metrics

( total_area_intersect, total_area_union, total_area_pred, total_area_target,

src/ksuit/utils/miou_utils.py:27

↓ 1 callersMethodtrack_after_accumulation_step

(self, **kwargs)

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

↓ 1 callersMethodtrack_after_update_step

(self, **kwargs)

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

↓ 1 callersMethodunscale_

src/ksuit/utils/amp_utils.py:83

↓ 1 callersFunctionunset_seed

src/ksuit/utils/seed.py:18

↓ 1 callersFunctionunzip

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

↓ 1 callersMethodupdate

(self, mapping)

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

↓ 1 callersFunctionvision_mamba

(input_shape, patch_size, dim, depth, drop_path_rate=0., **kwargs)

src/vislstm/external/models_mamba.py:805

↓ 1 callersFunctionwang_init_

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

vision_lstm/vision_lstm.py:42

↓ 1 callersMethodwas_applied_successfully

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

↓ 1 callersMethodwrap_compile

(self, ddp_model)

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

↓ 1 callersMethodwrap_ddp

(self, trainer_model)

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

↓ 1 callersMethodwrap_model

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

↓ 1 callersMethodzero_grad

(self, set_to_none=True)

src/ksuit/optim/optim_wrapper.py:242

src/ksuit/utils/checkpoint.py:184

src/vislstm/collators/mix_collator.py:74

src/vislstm/transforms/random_resized_crop.py:29

(self, x, ctx=None)

src/vislstm/transforms/three_augment.py:17

src/vislstm/transforms/resize.py:14

(self, x, ctx=None)

src/vislstm/transforms/color_jitter.py:34

(self, xseg, ctx=None)

src/vislstm/transforms/segmentation_random_crop.py:14

(self, xseg, ctx=None)

src/vislstm/transforms/segmentation_random_horizontal_flip.py:11

(self, xseg, ctx=None)

src/vislstm/transforms/segmentation_pad.py:13

(self, xseg, ctx=None)

src/vislstm/transforms/segmentation_random_resize.py:18

(self, x, ctx=None)

src/vislstm/transforms/rand_augment.py:90

(self, x, ctx=None)

src/vislstm/transforms/gaussian_blur_pil.py:18

(self, xseg, ctx=None)

src/vislstm/transforms/segmentation_resize.py:13

(self, module, module_input, module_output)

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

(self, module, grad_input, grad_output)

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

(self, module, module_input, module_output)

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

(self, module, module_input, module_output)

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

src/ksuit/data/collators/collator.py:8

src/ksuit/data/collators/compose_collator.py:8

(self, x, ctx=None)

src/ksuit/data/transforms/base/compose_transform.py:22

(self, x, ctx=None)

src/ksuit/data/transforms/base/transform.py:12

(self, x, ctx=None)

src/ksuit/data/transforms/base/stochastic_transform.py:49

(self, x, ctx=None)

src/ksuit/data/transforms/norm/base/norm_base.py:10

(self, features: list)

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

(self, features: list)

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

(self, features: list)

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

(self, features: list)

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

(self, _, __, output)

src/ksuit/models/extractors/base/forward_hook.py:13

Method__contains__

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

Method__contains__

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

Method__contains__

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

Method__contains__

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

Method__contains__

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

Method__contains__

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

Method__contains__

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

Method__contains__

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

Method__enter__

Method__enter__

src/ksuit/utils/amp_utils.py:70

Method__enter__

src/ksuit/utils/seed.py:34

Method__enter__

src/ksuit/utils/noop_tqdm.py:5

Method__enter__

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

src/ksuit/utils/checkpoint.py:84

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

(self, *args, **kwargs)

(self, *args, **kwargs)

src/ksuit/utils/amp_utils.py:73

(self, *_, **__)

src/ksuit/utils/seed.py:37

(self, *_, **__)

src/ksuit/utils/noop_tqdm.py:8

(self, *_, **__)

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

← previous next →401–500 of 1,346, ranked by callers