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

↓ 3 callersMethodget_stage_checkpoint_path

(self, stage_name: str, stage_id: str)

src/ksuit/providers/path_provider.py:61

↓ 3 callersFunctionget_total_cpu_count

()

src/ksuit/utils/num_worker_heuristic.py:74

↓ 3 callersMethodhas_same_specified_properties

(self, other)

src/ksuit/utils/checkpoint.py:100

↓ 3 callersFunctionhash_tensor_entries

(tensor, num_primes=None, shuffle_primes_seed=None)

src/ksuit/utils/tensor_hashing.py:58

↓ 3 callersMethodhigher_is_better

(self, key)

src/ksuit/providers/metric_property_provider.py:38

↓ 3 callersMethodinit_optim

(self, model)

src/ksuit/initializers/resume_initializer.py:33

↓ 3 callersMethodinit_weights

(self, model)

src/ksuit/initializers/resume_initializer.py:18

↓ 3 callersMethodinstantiate

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

src/ksuit/factory/model_factory.py:12

↓ 3 callersFunctionlog_from_all_ranks

()

src/ksuit/utils/logging_utils.py:47

↓ 3 callersMethodnormalize

(self, x, inplace=True)

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

↓ 3 callersFunctionseconds_to_duration_str

(total_seconds)

src/ksuit/utils/formatting_utils.py:82

↓ 3 callersMethodsegment

(self, x)

src/vislstm/models/single/vit.py:225

↓ 3 callersFunctionset_seed

(seed)

src/ksuit/utils/seed.py:9

↓ 3 callersMethodto_target_specification

removes all overly specified properties of self (depending on the specified properties of target) e.g. self=Checkpoint(epoch=

src/ksuit/utils/checkpoint.py:166

↓ 2 callersMethod__init__

(self, config: sLSTMCellConfig)

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

↓ 2 callersMethod__init__

Simple block wrapping a mixer class with LayerNorm/RMSNorm and residual connection" This Block has a slightly different structure co

src/vislstm/external/models_mamba.py:259

↓ 2 callersMethod__register_sampler_config

(self, dataset, mode, name, collator=None)

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

↓ 2 callersFunction_all_gather_nondistributed

(x, og_device)

src/ksuit/distributed/gather.py:114

↓ 2 callersMethod_get_ckpt_uri

(self, prefix, suffix)

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

↓ 2 callersMethod_get_hp_file_uri

(hp_file)

src/ksuit/configs/hyperparams.py:27

↓ 2 callersMethod_get_trainer_ckpt_file

(self)

src/ksuit/initializers/resume_initializer.py:58

↓ 2 callersMethod_getitem

(self, item, idx, ctx=None)

src/ksuit/data/wrappers/sample_wrappers/base/transform_wrapper_base.py:15

↓ 2 callersMethod_getitem_cached

(self, idx, ctx, item)

src/vislstm/sample_wrappers/segmentation_transform_wrapper.py:10

↓ 2 callersFunction_is_compatible

(device, dtype)

src/ksuit/utils/amp_utils.py:52

↓ 2 callersMethod_log

(self, key, value, logger=None, format_str=None)

src/ksuit/callbacks/base/writers/log_writer.py:43

↓ 2 callersMethod_log

(self, module, module_input, module_output, tensor_name)

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

↓ 2 callersMethod_remove_large_collections

(stage_hp)

src/ksuit/configs/hyperparams.py:91

↓ 2 callersMethod_save

(self, ckpt, model)

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

↓ 2 callersMethod_save_config_as_yaml

(self)

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

↓ 2 callersMethod_set_proj_up_dim

(self, embedding_dim: int)

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

↓ 2 callersMethod_shear_degrees

(self, magnitude)

src/vislstm/transforms/rand_augment.py:200

↓ 2 callersMethod_step

( self, total_loss, model, accumulation_steps, ite

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

↓ 2 callersMethod_to_ckpt_dict

(self, model, ckpt)

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

↓ 2 callersMethod_translation

(self, magnitude)

src/vislstm/transforms/rand_augment.py:250

↓ 2 callersMethod_update_state

(self, model, requires_grad)

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

↓ 2 callersMethodadd_item

(mode, item)

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

↓ 2 callersFunctionadd_stdout_handler

(prefix="")

src/ksuit/utils/logging_utils.py:24

↓ 2 callersMethodafter_epoch

(self, **kwargs)

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

↓ 2 callersMethodafter_update

(self, effective_batch_size, **kwargs)

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

↓ 2 callersMethodafter_update_step

after_update_step hook (e.g. for EMA)

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

↓ 2 callersFunctionall_ctor_kwarg_names

(cls, result=None)

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

↓ 2 callersFunctionall_reduce_mean_grad

(x)

src/ksuit/distributed/gather.py:156

↓ 2 callersFunctionall_reduce_mean_nograd

(x)

src/ksuit/distributed/gather.py:166

↓ 2 callersFunctionall_reduce_sum_grad

(x)

src/ksuit/distributed/gather.py:145

↓ 2 callersFunctionall_reduce_sum_nograd

(x)

src/ksuit/distributed/gather.py:140

↓ 2 callersMethodbarrier

(self)

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

↓ 2 callersMethodbefore_training

(self, model)

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

↓ 2 callersFunctionbias_linspace_init_

Linearly spaced bias init across dimensions.

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

↓ 2 callersMethodcheck_valid

(root)

src/vislstm/datasets/ade20k.py:50

↓ 2 callersFunctioncheck_versions

(verbose)

src/ksuit/utils/version_check.py:15

↓ 2 callersMethodclassify

(self, *args, **kwargs)

src/vislstm/models/single/vit.py:220

↓ 2 callersMethodclear_buffers

(self)

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

↓ 2 callersMethodcontains_checkpoint_string

(source)

src/ksuit/utils/checkpoint.py:128

↓ 2 callersFunctioncreate_feedforward

(config: FeedForwardConfig)

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

↓ 2 callersMethoddenormalize

(self, x, inplace=True)

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

↓ 2 callersMethoddevice

(self)

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

↓ 2 callersFunctiondevices_are_idle

(device_ids)

src/main_run_folder.py:31

↓ 2 callersMethodfind_checkpoint_string

(source)

src/ksuit/utils/checkpoint.py:133

↓ 2 callersMethodfinish

()

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

↓ 2 callersMethodforward

(self, x, mask_generator=None, idx=None)

src/vislstm/models/single/vit.py:182

↓ 2 callersMethodforward

(self, x, mask_generator=None, idx=None)

src/vislstm/models/single/vislstm.py:206

↓ 2 callersMethodforward

(self, x)

src/vislstm/models/single/vismamba.py:73

↓ 2 callersMethodfrom_checkpoint_string

(checkpoint_string)

src/ksuit/utils/checkpoint.py:121

↓ 2 callersMethodfrom_cli_args

()

src/ksuit/configs/cli_args.py:32

↓ 2 callersMethodfrom_uri

(uri: str, template_path="./setup")

src/ksuit/configs/static_config.py:16

↓ 2 callersFunctionget_backend

(accelerator, device_ids=None)

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

↓ 2 callersFunctionget_fair_cpu_count

()

src/ksuit/utils/num_worker_heuristic.py:5

↓ 2 callersFunctionget_frozen_param_count

(model)

src/ksuit/utils/model_utils.py:63

↓ 2 callersMethodget_local_rank

(self)

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

↓ 2 callersFunctionget_managed_world_size

()

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

↓ 2 callersMethodget_num_nodes

(self)

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

↓ 2 callersMethodget_params

(self, height, width)

src/vislstm/transforms/segmentation_random_crop.py:33

↓ 2 callersFunctionget_parser

()

src/main_sbatch.py:14

↓ 2 callersMethodget_primitive_config_uri

(self, stage_name: str, stage_id: str)

src/ksuit/providers/path_provider.py:40

↓ 2 callersMethodget_primitive_summary_uri

(self, stage_name: str, stage_id: str)

src/ksuit/providers/path_provider.py:54

↓ 2 callersMethodget_random_bbox

(self, h, w, lamb)

src/vislstm/collators/mix_collator.py:170

↓ 2 callersMethodget_weight_decay_optim_groups

Return a tuple of two sequences, one for parameters with weight decay and one for parameters without weight decay. Performs checks to ensure t

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

↓ 2 callersFunctionhash_rgb

(tensor, dim=1)

src/ksuit/utils/tensor_hashing.py:70

↓ 2 callersFunctionis_bfloat16_compatible

(device)

src/ksuit/utils/amp_utils.py:61

↓ 2 callersFunctionis_float16_compatible

(device)

src/ksuit/utils/amp_utils.py:65

↓ 2 callersMethodis_local_rank0

(self)

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

↓ 2 callersFunctionjoin_names

(name1, name2)

src/ksuit/utils/naming_utils.py:1

↓ 2 callersMethodkeys

(self)

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

↓ 2 callersMethodload_state_dict

(self, state_dict)

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

↓ 2 callersMethodnormalize

(self, x, inplace=True)

src/ksuit/data/transforms/norm/image_range_norm.py:8

↓ 2 callersMethodnormalize

(self, x, inplace=True)

src/ksuit/data/transforms/norm/image_moment_norm.py:14

↓ 2 callersFunctionrun_managed

(main, accelerator="gpu", devices=None)

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

↓ 2 callersFunctionrun_unmanaged

(main, devices, accelerator="gpu", master_port=None, mig_devices=None)

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

↓ 2 callersMethodset_forward

(self, key, value)

src/ksuit/utils/bidict.py:25

↓ 2 callersMethodshould_log_after_epoch

(self, checkpoint)

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

↓ 2 callersMethodshould_log_after_sample

(self, checkpoint, effective_batch_size)

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

↓ 2 callersFunctionsmall_init_

Fills the input Tensor with values according to the method described in Transformers without Tears: Improving the Normalization of Self-Atten

vision_lstm/vision_lstm.py:31

↓ 2 callersFunctionsmall_init_

Fills the input Tensor with values according to the method described in Transformers without Tears: Improving the Normalization of Self-Atten

vision_lstm/vision_lstm2.py:31

↓ 2 callersMethodstate_dict

(self)

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

↓ 2 callersMethodstep

(optimizer, *args, **kwargs)

src/ksuit/utils/amp_utils.py:87

↓ 2 callersMethodto_fully_specified

(self, updates_per_epoch, effective_batch_size)

src/ksuit/utils/checkpoint.py:40

↓ 2 callersMethodto_fully_specified_from_fnames

(ckpt_folder, ckpt, prefix=None, suffix=None)

src/ksuit/utils/checkpoint.py:145

↓ 2 callersFunctionto_ntuple

(x, n)

src/ksuit/utils/param_checking.py:24

↓ 2 callersFunctiontype_from_name

tries to import type_name from any of the modules identified by module_names e.g. module_names=[loss_functions, torch.nn] type_name=bce_loss

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

↓ 2 callersMethodupdate

(self, *args, **kwargs)

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

← previous next →101–200 of 1,346, ranked by callers