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

↓ 1 callersMethodget_backward

(self, key)

src/ksuit/utils/bidict.py:22

↓ 1 callersFunctionget_bool_gather_supported

()

src/ksuit/distributed/gather.py:25

↓ 1 callersFunctionget_cli_command

()

src/ksuit/utils/system_info.py:13

↓ 1 callersMethodget_config_of_previous_stage

(self, stage_name, stage_id)

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

↓ 1 callersMethodget_data_loader

(self, periodic_callbacks, batch_size)

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

↓ 1 callersMethodget_data_source_mode

(self, dataset_identifier)

src/ksuit/providers/dataset_config_provider.py:38

↓ 1 callersMethodget_default_callbacks

(self)

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

↓ 1 callersFunctionget_denorm_transform

(transform, inplace=False)

src/ksuit/utils/transform_utils.py:29

↓ 1 callersFunctionget_device_and_bfloat16supported

()

src/ksuit/distributed/gather.py:10

↓ 1 callersMethodget_forward

(self, key)

src/ksuit/utils/bidict.py:19

↓ 1 callersMethodget_global_dataset_path

(self, dataset_identifier)

src/ksuit/providers/dataset_config_provider.py:29

↓ 1 callersFunctionget_grad_scaler_and_autocast_context

(precision, device)

src/ksuit/utils/amp_utils.py:95

↓ 1 callersFunctionget_installed_cuda_version

()

src/ksuit/utils/system_info.py:21

↓ 1 callersMethodget_interval_string_verbose

(self)

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

↓ 1 callersMethodget_local_rank

(self)

src/ksuit/distributed/config/default_config.py:13

↓ 1 callersMethodget_main_sampler

( self, train_dataset, num_repeats=1, shuffle=True, )

src/ksuit/utils/data_container.py:58

↓ 1 callersFunctionget_managed_rank

()

src/ksuit/distributed/config/__init__.py:28

↓ 1 callersMethodget_managed_rank

extract rank from environment variables (e.g. via SLURM_PROCID) required for managed runs to derive rank for initializing process gro

src/ksuit/distributed/config/base_config.py:21

↓ 1 callersMethodget_managed_world_size

extract world_size from environment variables (e.g. via SLURM_NTASKS_PER_NODE * SLURM_JOB_NUM_NODES) required for managed runs to der

src/ksuit/distributed/config/base_config.py:14

↓ 1 callersMethodget_model_kwargs

(self)

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

↓ 1 callersFunctionget_nograd_paramnames

(model)

src/ksuit/utils/model_utils.py:7

↓ 1 callersFunctionget_norm_transform

(transform)

src/ksuit/utils/transform_utils.py:20

↓ 1 callersMethodget_num_nodes

(self)

src/ksuit/distributed/config/base_config.py:11

↓ 1 callersMethodget_param_group_modifiers

()

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

↓ 1 callersMethodget_params

(self, img)

src/vislstm/transforms/random_resized_crop.py:33

↓ 1 callersMethodget_params

(self)

src/vislstm/transforms/color_jitter.py:47

↓ 1 callersMethodget_params

(self)

src/vislstm/transforms/segmentation_random_resize.py:47

↓ 1 callersMethodget_params

(self)

src/vislstm/transforms/gaussian_blur_pil.py:26

↓ 1 callersFunctionget_powers_of_two

(min_value, max_value)

src/ksuit/utils/math_utils.py:4

↓ 1 callersMethodget_primitive_entries_uri

(self, stage_name: str, stage_id: str)

src/ksuit/providers/path_provider.py:47

↓ 1 callersMethodget_properties

(self, model, name, param)

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

↓ 1 callersMethodget_stage_hp

( hp_file, testrun=False, minmodelrun=False, mindatarun=False,

src/ksuit/configs/hyperparams.py:46

↓ 1 callersMethodget_start_checkpoint

(self)

src/ksuit/initializers/resume_initializer.py:61

↓ 1 callersMethodget_summary_of_previous_stage

(self, stage_name, stage_id)

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

↓ 1 callersFunctionget_supported_precision

(desired_precision, device, backup_precision=None)

src/ksuit/utils/amp_utils.py:12

↓ 1 callersMethodget_trainer_callbacks

(self, model=None)

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

↓ 1 callersMethodget_trainer_model

(self, model)

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

↓ 1 callersMethodget_wrapper_of_type

(self, wrapper_type)

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

↓ 1 callersMethodget_wrappers_of_type

(self, wrapper_type)

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

↓ 1 callersMethodgetall_class

(self)

src/vislstm/datasets/imagenet1k.py:25

↓ 1 callersMethodgetitem_class

(self, idx)

src/ksuit/datasets/random_dataset.py:25

↓ 1 callersMethodgetitem_segmentation

(self, idx, ctx=None)

src/vislstm/sample_wrappers/segmentation_transform_wrapper.py:35

↓ 1 callersMethodgetitem_x

(self, idx, ctx=None)

src/vislstm/sample_wrappers/segmentation_transform_wrapper.py:32

↓ 1 callersMethodgetitem_x

(self, idx)

src/ksuit/datasets/random_dataset.py:22

↓ 1 callersMethodgetshape

(self, kind)

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

↓ 1 callersMethodgetshape_class

(self)

src/ksuit/datasets/random_dataset.py:19

↓ 1 callersFunctiongetshape_instead_of_getdim

(getdim_names)

src/ksuit/data/error_messages.py:1

↓ 1 callersMethodhas_item

(mode, item)

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

↓ 1 callersMethodinit_callbacks

(self, callbacks, model)

src/ksuit/initializers/resume_initializer.py:84

↓ 1 callersMethodinit_trainer

(self, trainer)

src/ksuit/initializers/resume_initializer.py:79

↓ 1 callersFunctioninit_wandb

( device: str, run_name: str, stage_hp: dict, wandb_config: WandbConfig,

src/ksuit/utils/wandb_utils.py:23

↓ 1 callersMethodinitialize

(self, lr_scale_factor=None)

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

↓ 1 callersMethodinitialize_optim

(self, lr_scale_factor=None)

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

↓ 1 callersMethodinitialize_weights

(self)

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

↓ 1 callersMethodinstance

(cls, config: sLSTMCellConfig)

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

↓ 1 callersMethodinstantiate

(self, kind, optional_kwargs=None, **kwargs)

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

↓ 1 callersFunctionintersect_and_union

(pred, target, num_classes: int, ignore_index: int)

src/ksuit/utils/miou_utils.py:4

↓ 1 callersFunctionis_data_rank0

()

src/ksuit/distributed/config/__init__.py:44

↓ 1 callersMethodis_data_rank0

(self)

src/ksuit/distributed/config/base_config.py:42

↓ 1 callersFunctionis_local_rank0

()

src/ksuit/distributed/config/__init__.py:52

↓ 1 callersMethodis_managed

(self)

src/ksuit/distributed/config/base_config.py:5

↓ 1 callersMethodis_neutral_key

(self, key)

src/ksuit/providers/metric_property_provider.py:35

↓ 1 callersMethodis_own_work

(self, idx)

src/ksuit/distributed/config/base_config.py:62

↓ 1 callersFunctionis_power_of_two

(value)

src/ksuit/utils/math_utils.py:11

↓ 1 callersMethodis_rank0

(self)

src/ksuit/distributed/config/base_config.py:52

↓ 1 callersFunctionlog_device_info

(accelerator, device_ids)

src/ksuit/distributed/run/utils.py:94

↓ 1 callersFunctionlog_distributed_config

()

src/ksuit/distributed/config/__init__.py:68

↓ 1 callersMethodlog_distributed_config

(self)

src/ksuit/distributed/config/base_config.py:70

↓ 1 callersMethodlog_stage_hp

(stage_hp)

src/ksuit/configs/hyperparams.py:98

↓ 1 callersFunctionlog_system_info

()

src/ksuit/utils/system_info.py:29

↓ 1 callersFunctionmain

(data, name, device, batch_size, num_workers, precision)

eval.py:31

↓ 1 callersFunctionmain

()

logs/clean_logs.py:5

↓ 1 callersFunctionmain

( nodes, gpus, time, account, qos, constraint, preload

src/main_sbatch.py:35

↓ 1 callersFunctionmain

( folder, devices, accelerator, wandb_config, wandb_mode, fore

src/main_run_folder.py:45

↓ 1 callersMethodmatch_pattern

(self, string: str, pattern: str)

src/ksuit/pattern_matchers/base/pattern_matcher.py:2

↓ 1 callersFunctionmulticlass_global_to_local

( global_root, local_root, num_workers=0, log_fn=None, )

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

↓ 1 callersFunctionmulticlass_knn

( train_x, test_x, train_y, test_y, k=10, tau=0.07, ba

src/ksuit/metrics/multiclass_knn.py:80

↓ 1 callersMethodnext_epoch

(self)

src/ksuit/utils/update_counter.py:64

↓ 1 callersMethodnext_update

(self)

src/ksuit/utils/update_counter.py:67

↓ 1 callersMethodoptim_schedule_step

(self)

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

↓ 1 callersMethodoptim_step

(self, grad_scaler)

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

↓ 1 callersMethodoptim_zero_grad

(self, set_to_none=True)

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

↓ 1 callersFunctionparallel_stabilized_simple

This is the mLSTM cell in parallel form. This version is stabilized. We control the range of exp() arguments by ensuring that they are al

vision_lstm/vision_lstm.py:49

↓ 1 callersFunctionparallel_stabilized_simple

This is the mLSTM cell in parallel form. This version is stabilized. We control the range of exp() arguments by ensuring that they are al

vision_lstm/vision_lstm2.py:49

↓ 1 callersFunctionparse_args

()

eval.py:12

↓ 1 callersFunctionparse_args

()

src/main_run_folder.py:16

↓ 1 callersFunctionparse_devices

(accelerator, devices, mig_devices=None)

src/ksuit/distributed/run/utils.py:55

↓ 1 callersFunctionpascal_to_snake

convert pascal/camel to snake case https://learn.microsoft.com/en-us/visualstudio/code-quality/ca1709?view=vs-2022 "By convention, two-letter

src/ksuit/utils/naming_utils.py:8

↓ 1 callersMethodregister_root_datasets

(self, dataset_config_provider=None, is_mindatarun=False)

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

↓ 1 callersMethodregister_sampler_configs

(self, trainer)

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

↓ 1 callersMethodreset_parameters

(self)

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

↓ 1 callersMethodreset_parameters

(self, **kwargs)

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

↓ 1 callersMethodreset_parameters

(self)

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

↓ 1 callersMethodreset_parameters

(self, **kwargs)

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

↓ 1 callersMethodreset_parameters

(self)

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

↓ 1 callersMethodreset_parameters

(self)

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

↓ 1 callersMethodreset_parameters

(self)

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

↓ 1 callersMethodreset_parameters

(self)

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

↓ 1 callersMethodreset_parameters

Resets this layer's parameters to their initial values.

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

↓ 1 callersMethodreset_parameters

(self)

vision_lstm/vision_lstm.py:154

← previous next →301–400 of 1,346, ranked by callers