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

↓ 52 callersMethodget

(key)

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

↓ 44 callersMethodto

(self, device, *args, **kwargs)

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

↓ 35 callersMethodadd_scalar

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

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

↓ 27 callersMethodkeys

(self)

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

↓ 21 callersMethodcreate

(kwargs)

src/ksuit/utils/checkpoint.py:10

↓ 20 callersMethodset

(key, value)

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

↓ 20 callersFunctionto_path

(path, mkdir=False, should_exist=True, check_exists=True, suffix=None)

src/ksuit/utils/param_checking.py:75

↓ 19 callersFunctionis_distributed

()

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

↓ 18 callersMethodsave

( self, model, checkpoint, trainer=None, save_weig

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

↓ 17 callersFunctionis_rank0

()

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

↓ 15 callersMethod_log_and_print

(self, msg)

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

↓ 15 callersMethod_log_and_print

(self, msg)

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

↓ 14 callersMethodget_dataset

(self, key=None, mode=None, max_size=None, shuffle_seed=None, return_type=dict)

src/ksuit/utils/data_container.py:47

↓ 13 callersMethodclone

(self)

src/vislstm/modules/xlstm/components/util.py:59

↓ 12 callersMethod_log_and_print

(self, msg)

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

↓ 12 callersFunctionall_gather_nograd

(x)

src/ksuit/distributed/gather.py:97

↓ 11 callersFunctionget_world_size

()

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

↓ 11 callersFunctionlog

(log_fn, msg)

src/ksuit/utils/logging_utils.py:8

↓ 10 callersFunctionbarrier

()

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

↓ 9 callersFunction_is_ntuple

(n)

src/ksuit/utils/param_checking.py:17

↓ 9 callersFunction_ntuple

(n)

src/ksuit/utils/param_checking.py:7

↓ 9 callersMethod_register_sampler_config_from_key

(self, key, mode, max_size=None)

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

↓ 9 callersMethoditerate_over_dataset

( self, forward_fn, config_id, batch_size, data_it

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

↓ 8 callersMethodclear

()

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

↓ 8 callersMethoddevice

(self)

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

↓ 8 callersMethodget_rank

(self)

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

↓ 8 callersMethodget_roots

(self, global_root, identifier, local_root=None)

src/ksuit/providers/dataset_config_provider.py:15

↓ 8 callersFunctionlist_to_string

(tensor)

src/ksuit/utils/formatting_utils.py:47

↓ 8 callersMethodload_state_dict

(self, state_dict, strict=True)

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

↓ 8 callersMethodto_short_interval_string

(self)

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

↓ 8 callersMethodupdate

()

src/ksuit/utils/amp_utils.py:91

↓ 7 callersMethod__init__

( self, dim, direction, expansion=2, qkv_block_siz

vision_lstm/vision_lstm2.py:353

↓ 7 callersMethodcreate

(self, obj_or_kwargs, collate_fn=None, **kwargs)

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

↓ 7 callersMethodcreate_list

(self, collection, collate_fn=None, **kwargs)

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

↓ 7 callersFunctionget_rank

()

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

↓ 7 callersMethodgetdim

(self, kind)

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

↓ 7 callersFunctioninterpolate_sincos

(embed, seqlens, mode="bicubic")

vision_lstm/vision_lstm_util.py:29

↓ 7 callersMethodload_state_dict

(self, state_dict_to_load)

src/ksuit/optim/optim_wrapper.py:252

↓ 7 callersMethodlog

(self)

src/ksuit/configs/cli_args.py:77

↓ 7 callersFunctionselect_with_path

(obj, path)

src/ksuit/utils/select_with_path.py:1

↓ 7 callersFunctionsmall_init_init_

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

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

↓ 6 callersMethod__init__

( self, dim, direction, expansion=2, qkv_block_siz

vision_lstm/vision_lstm.py:353

↓ 6 callersMethodapply

(x)

src/ksuit/distributed/functional/all_gather_grad_overwrite.py:8

↓ 6 callersMethoddisable_hooks

(self)

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

↓ 6 callersMethodget_backend

(self)

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

↓ 6 callersFunctionget_rng_from_global

()

src/ksuit/data/utils/random.py:5

↓ 6 callersMethodget_world_size

(self)

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

↓ 6 callersFunctionis_managed

()

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

↓ 6 callersFunctionshort_number_str

(number, precision=1)

src/ksuit/utils/formatting_utils.py:7

↓ 5 callersMethod_get_modelname_and_ckpturi

(self, file_type, model=None, model_name=None)

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

↓ 5 callersFunction_prepare_tensor

prepare for distributed communication - wrap primitive types into tensors - push tensor onto supported device - convert bool to float

src/ksuit/distributed/gather.py:35

↓ 5 callersMethod_sample_interpolation

(self)

src/vislstm/transforms/rand_augment.py:98

↓ 5 callersFunctionfloat_to_scientific_notation

(value, max_precision, remove_plus=True)

src/ksuit/utils/formatting_utils.py:65

↓ 5 callersMethodflush

()

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

↓ 5 callersMethodget_default_callback_kwargs

(self)

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

↓ 5 callersFunctionget_local_rank

()

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

↓ 5 callersMethodget_stage_output_path

(self, stage_name: str, stage_id: str, mkdir=True)

src/ksuit/providers/path_provider.py:16

↓ 5 callersMethodis_distributed

()

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

↓ 5 callersFunctionmain

()

src/main_train.py:4

↓ 5 callersMethodmatch_patterns

(self, string: str, patterns: list[str])

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

↓ 5 callersFunctionoptional_ctx

(fn, ctx)

src/ksuit/data/utils/optional_kwargs.py:4

↓ 5 callersMethodreset_parameters

(self)

vision_lstm/vision_lstm.py:454

↓ 5 callersMethodreset_parameters

(self)

vision_lstm/vision_lstm2.py:478

↓ 5 callersMethodshuffle

(self, item, permutation)

src/vislstm/collators/mix_collator.py:189

↓ 5 callersMethodstate_dict

(self)

src/ksuit/optim/optim_wrapper.py:247

↓ 4 callersMethod_adjust_factor

(self, magnitude)

src/vislstm/transforms/rand_augment.py:152

↓ 4 callersMethod_check_bool

(self, key, default)

src/ksuit/configs/static_config.py:28

↓ 4 callersMethod_log

(self, module, module_input, module_output, tensor_name, invalid_value)

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

↓ 4 callersMethod_log

(self, module, module_input, module_output, tensor_name, invalid_value)

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

↓ 4 callersFunctionaccelerator_to_device

(accelerator)

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

↓ 4 callersMethodcopy

(self)

src/ksuit/utils/checkpoint.py:22

↓ 4 callersMethodcreate_dict

(self, collection, collate_fn=None, **kwargs)

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

↓ 4 callersMethoddisable_hooks

(self)

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

↓ 4 callersMethodenable_hooks

(self)

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

↓ 4 callersMethodextract

(self)

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

↓ 4 callersFunctionget_named_models

(model)

src/ksuit/utils/model_utils.py:38

↓ 4 callersMethodget_output_shape

(self, input_shape)

src/vislstm/poolings/bilateral.py:11

↓ 4 callersMethodget_primitive_output_path

(self, stage_name: str, stage_id: str)

src/ksuit/providers/path_provider.py:32

↓ 4 callersFunctionget_trainable_param_count

(model)

src/ksuit/utils/model_utils.py:59

↓ 4 callersMethodget_value

(self)

src/ksuit/utils/schedule_wrapper.py:11

↓ 4 callersMethodregister_hooks

(self, model)

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

↓ 4 callersFunctionsnake_type_name

(obj)

src/ksuit/utils/naming_utils.py:36

↓ 4 callersMethodstate_dict

(self)

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

↓ 4 callersFunctionto_ntuple

(x, n)

vision_lstm/vision_lstm_util.py:24

↓ 4 callersMethodtrain

(self, model, callbacks=None)

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

↓ 3 callersMethod__init__

( self, *args, drop_prob: float = 0., scale_by_keep: bool = Tr

vision_lstm/vision_lstm_util.py:169

↓ 3 callersFunction_add_handler

(handler, prefix="")

src/ksuit/utils/logging_utils.py:12

↓ 3 callersFunction_all_gather_grad

(x, all_gather_fn, batch_dim=0)

src/ksuit/distributed/gather.py:68

↓ 3 callersFunction_check_pip_dependency

(actual_version, expected_version, pip_dependency_name)

src/ksuit/utils/version_check.py:41

↓ 3 callersMethod_mkdir

(path: Path)

src/ksuit/providers/path_provider.py:12

↓ 3 callersMethod_periodic_callback

(self, interval_type, **kwargs)

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

↓ 3 callersMethod_process_checkpoint

(self, parent, parent_accessor)

src/ksuit/configs/resolver_processors/minduration_postprocessor.py:50

↓ 3 callersMethod_should_stop

(self)

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

↓ 3 callersFunctionadd_global_handlers

(log_file_uri=None)

src/ksuit/utils/logging_utils.py:30

↓ 3 callersMethodadd_module_name

(self, module_name, index=None)

src/ksuit/factory/base/single_factory.py:17

↓ 3 callersFunctionall_gather_nograd_clipped

(x, max_length)

src/ksuit/distributed/gather.py:121

↓ 3 callersFunctioncheck_single_device_visible

(accelerator)

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

↓ 3 callersMethodenable_hooks

(self)

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

↓ 3 callersMethodget_default_callback_intervals

(self)

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

↓ 3 callersFunctionget_num_nodes

()

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

next →1–100 of 1,346, ranked by callers