MCPcopy Create free account

hub / github.com/ZZongzheng0918/TA-VLA / functions

Functions459 in github.com/ZZongzheng0918/TA-VLA

⨍ Functions 459◇ Types & classes 131
Methodsetup
(self)
src/openpi/models/gemma_fast.py:242
Methodsetup
(self)
src/openpi/models/lora.py:43
Methodsetup
(self)
src/openpi/models/lora.py:96
Functionsetup_master_bot
(bot)
examples/aloha_real/robot_utils.py:221
Functionsetup_puppet_bot
(bot)
examples/aloha_real/robot_utils.py:214
Functionstate_map
Apply a function to the leaves of the state that match the filter.
src/openpi/shared/nnx_utils.py:66
Methodstep
(carry)
src/openpi/models/pi0.py:394
Methodstep
(carry)
src/openpi/models/pi0_fast.py:267
Functionsync_puppet_to_master
(master_bot_left, master_bot_right, puppet_bot_left, puppet_bot_right)
examples/aloha_real/robot_utils.py:248
Functiontest_absolute_actions
()
src/openpi/transforms_test.py:42
Functiontest_absolute_actions_noop
()
src/openpi/transforms_test.py:52
Functiontest_broker
()
src/openpi/policies/policy_test.py:21
Functiontest_delta_actions
()
src/openpi/transforms_test.py:19
Functiontest_delta_actions_noop
()
src/openpi/transforms_test.py:29
Functiontest_download_fsspec
()
src/openpi/shared/download_test.py:47
Functiontest_download_local
(tmp_path: pathlib.Path)
src/openpi/shared/download_test.py:16
Functiontest_download_s3
()
src/openpi/shared/download_test.py:37
Functiontest_download_s3_dir
()
src/openpi/shared/download_test.py:27
Functiontest_extract_prompt_from_task
()
src/openpi/transforms_test.py:114
Functiontest_fast_tokenizer
()
src/openpi/models/tokenizer_test.py:14
Functiontest_infer
()
src/openpi/policies/policy_test.py:10
Functiontest_lora_einsum_params_shape
()
src/openpi/models/lora_test.py:8
Functiontest_lora_einsum_same_output
()
src/openpi/models/lora_test.py:34
Functiontest_lora_ffn_params_shape
()
src/openpi/models/lora_test.py:53
Functiontest_lora_ffn_same_output
()
src/openpi/models/lora_test.py:77
Functiontest_make_bool_mask
()
src/openpi/transforms_test.py:65
Functiontest_model_restore
()
src/openpi/models/model_test.py:79
Functiontest_normalize_update
()
src/openpi/shared/normalize_test.py:6
Functiontest_pack_unpack
(data)
packages/openpi-client/src/openpi_client/msgpack_numpy_test.py:42
Functiontest_pi0_action_expert_lora
()
src/openpi/models/pi0_test.py:29
Functiontest_pi0_all_lora
()
src/openpi/models/pi0_test.py:40
Functiontest_pi0_fast_lora_model
()
src/openpi/models/model_test.py:57
Functiontest_pi0_fast_model
()
src/openpi/models/model_test.py:42
Functiontest_pi0_full_finetune
()
src/openpi/models/pi0_test.py:14
Functiontest_pi0_gemma_lora
()
src/openpi/models/pi0_test.py:20
Functiontest_pi0_lora_model
()
src/openpi/models/model_test.py:27
Functiontest_pi0_model
()
src/openpi/models/model_test.py:12
Functiontest_repack_transform
()
src/openpi/transforms_test.py:8
Functiontest_resize_with_pad_shapes
()
packages/openpi-client/src/openpi_client/image_tools_test.py:6
Functiontest_resize_with_pad_shapes
()
src/openpi/shared/image_tools_test.py:6
Functiontest_serialize_deserialize
()
src/openpi/shared/normalize_test.py:18
Functiontest_tokenize
()
src/openpi/models/tokenizer_test.py:6
Functiontest_tokenize_no_prompt
()
src/openpi/transforms_test.py:81
Functiontest_tokenize_prompt
()
src/openpi/transforms_test.py:70
Functiontest_torch_data_loader
()
src/openpi/training/data_loader_test.py:10
Functiontest_torch_data_loader_infinite
()
src/openpi/training/data_loader_test.py:26
Functiontest_torch_data_loader_parallel
()
src/openpi/training/data_loader_test.py:37
Functiontest_train
(tmp_path: pathlib.Path, config_name: str)
scripts/train_test.py:15
Functiontest_transform_dict
()
src/openpi/transforms_test.py:88
Functiontest_with_fake_dataset
()
src/openpi/training/data_loader_test.py:50
Functiontest_with_real_dataset
()
src/openpi/training/data_loader_test.py:65
Methodtorch_loader
(self)
src/openpi/training/data_loader.py:271
Functiontrain_step
( config: _config.TrainConfig, rng: at.KeyArrayLike, state: training_utils.TrainState, batch:
scripts/train.py:136
Methodtrainable_filter
Get the filter for the trainable parameters.
src/openpi/training/config.py:590
Functiontransform_dict
Transform the structure of a nested dictionary using a set of patterns. The transformation is defined using the `patterns` dictionary. The keys a
src/openpi/transforms.py:323
Functiontypecheck
(t: T)
src/openpi/shared/array_typing.py:50
Functionunpack_array
(obj)
packages/openpi-client/src/openpi_client/msgpack_numpy.py:43
Functionupdate_progress
(size: int)
src/openpi/shared/download.py:216
Functionwrapper
(*args: P.args, **kwargs: P.kwargs)
src/openpi/shared/nnx_utils.py:40
← previous401–459 of 459, ranked by callers