Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/apple/ml-llarp
/ functions
Functions
462 in github.com/apple/ml-llarp
⨍
Functions
462
◇
Types & classes
125
↳
Endpoints
3
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:111
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:130
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:146
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:172
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:210
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:259
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:308
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:340
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:367
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:387
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:406
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:434
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:470
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:508
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:570
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:597
Method
_get_uuid
(*args, **kwargs)
llarp/task/measures.py:674
Function
_recur_replace
(expr, search, replace)
llarp/task/utils.py:37
Method
access_idx
(k, name_to_idx)
llarp/dataset/episodes.py:65
Method
action_space
(self)
llarp/task/actions.py:23
Method
action_space
(self)
llarp/task/actions.py:90
Method
action_space
(self)
llarp/task/actions.py:149
Method
after_update
Copy over information from the end of the current rollout buffer into the rollout buffer start.
llarp/policies/transformer_storage.py:151
Method
all_cls
(self)
llarp/task/predicate_task.py:148
Method
att_reduce_mean
(loss)
llarp/trainer/transformer_ppo.py:211
Method
base_model
(self)
llarp/policies/utils.py:43
Method
base_model
(self)
llarp/policies/utils.py:166
Function
cat_demo_datasets
(demos: List[DemoDataset])
llarp/dataset/demo_dataset.py:71
Function
check_extracted_batch
Checks a batch is good. Note that the entire batch should be valid as we are going to attend over the entire sequence. Only the later stack p
llarp/policies/transformer_storage.py:349
Method
config
(self)
llarp/policies/utils.py:70
Method
config
(self)
llarp/policies/utils.py:162
Method
construct_envs
( self, config: "DictConfig", workers_ignore_signals: bool = False, enforce_sc
llarp/trainer/custom_env_factory.py:155
Method
construct_envs
( self, config, workers_ignore_signals: bool = False, enforce_scenes_greater_e
llarp/trainer/test_env_factory.py:126
Method
create_custom_forward
(module)
llarp/policies/llama_parallel.py:166
Method
custom_forward
(*inputs)
llarp/policies/llama_parallel.py:167
Method
d_model
(self)
llarp/policies/utils.py:53
Method
d_model
(self)
llarp/policies/utils.py:180
Method
decode
( self, non_causal_tokens, causal_embeds, att_masks, kv_cache=None,
llarp/policies/utils.py:213
Method
embed_action
(self, action, llm)
llarp/policies/action_decoders.py:54
Method
encode_name_dict
(d, name_to_idx)
llarp/dataset/episodes.py:72
Function
entity_matches
(entity: PddlEntity, match_entities: List[str])
llarp/dataset/dataset_validator.py:65
Method
evaluate_actions
( self, observations, rnn_hidden_states, prev_actions, masks,
llarp/policies/hl_policy.py:221
Method
evaluate_actions
( self, observations, action, )
llarp/policies/llm_policy.py:118
Method
evaluate_agent
( self, agent, envs, config, checkpoint_index, step_id,
llarp/trainer/custom_evaluator.py:43
Function
find_action
(search, all_actions)
llarp/dataset/dataset_validator.py:99
Method
forward
Must return shape [batch_size, # visual embeds, token embed dim]
llarp/policies/visual_encoders.py:18
Method
forward
(self, obs)
llarp/policies/visual_encoders.py:56
Method
forward
Always returns float32 data type regardless of net internal type.
llarp/policies/vis_bridge.py:58
Method
forward
(self, vis_features, obs)
llarp/policies/vis_bridge.py:123
Method
forward
( self, hidden_state, obs, embed_obs )
llarp/policies/action_decoders.py:43
Method
forward
(self, hidden_state, obs, embed_obs)
llarp/policies/action_decoders.py:113
Method
forward
( self, input_ids: torch.LongTensor = None, attention_mask: Optional[torch.Tensor] = N
llarp/policies/llama_parallel.py:53
Method
forward_norm
(self, x, obs)
llarp/policies/llm_policy.py:192
Function
freeze_params
(module)
llarp/policies/utils.py:74
Method
from_config
(cls, actor_critic: NetPolicy, config)
llarp/trainer/transformer_ppo.py:49
Method
from_json
(self, json_str: str, scenes_dir: Optional[str] = None)
llarp/dataset/episodes.py:164
Method
generate
(self, non_causal_tokens, causal_tokens, tokenizer, num_new_tokens)
llarp/policies/utils.py:283
Function
generate_episodes
(args, cfg, conn, iter_eps, proc_idx)
llarp/dataset/create_episodes.py:43
Function
get_base_info
(cfg, args, conn)
llarp/dataset/create_episodes.py:103
Method
get_distrib
(self, logits, **kwargs)
llarp/policies/action_decoders.py:57
Method
get_distrib
(self, logits, **kwargs)
llarp/policies/action_decoders.py:124
Method
get_episode_iterator
(self, *args, **kwargs)
llarp/dataset/episodes.py:173
Method
get_image_processor
(self)
llarp/policies/utils.py:65
Method
get_next_skill
( self, observations, rnn_hidden_states, prev_actions, masks,
llarp/policies/hl_policy.py:268
Method
get_num_rnn_layers
(self)
llarp/policies/cores/decoder.py:230
Method
get_observation
(self, *args, **kwargs)
llarp/task/sensors.py:39
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:78
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:122
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:144
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:171
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:212
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:250
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:340
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:366
Method
get_observation
(self, *args, task, **kwargs)
llarp/task/sensors.py:400
Method
get_policy_components
(self)
llarp/policies/hl_policy.py:203
Method
get_trainable_params
(self)
llarp/policies/cores/decoder.py:218
Method
get_value
(self, observations, hidden_states, prev_actions, masks)
llarp/policies/llm_policy.py:109
Method
get_vision_output_shape
(self)
llarp/policies/utils.py:56
Method
get_visual_encoder
(self)
llarp/policies/utils.py:62
Method
get_visual_pooler
(self)
llarp/policies/utils.py:59
Method
goal_expr
(self)
llarp/task/predicate_task.py:198
Method
hidden_size
(self)
llarp/policies/action_decoders.py:50
Method
hidden_size
(self)
llarp/policies/action_decoders.py:110
Method
hidden_window_dim
(self)
llarp/policies/cores/decoder.py:201
Method
hidden_window_dim
(self)
llarp/policies/cores/base_core.py:39
Method
llm_head
(self)
llarp/policies/utils.py:48
Method
llm_head
(self)
llarp/policies/utils.py:176
Function
log_ep_counts
Used for debugging.
llarp/dataset/generator.py:725
Method
n_toks_per_step
(self)
llarp/policies/cores/decoder.py:493
Method
num_recurrent_layers
(self)
llarp/policies/hl_policy.py:331
Method
num_recurrent_layers
(self)
llarp/policies/llm_policy.py:168
Method
num_tokens
(self)
llarp/policies/vis_bridge.py:22
Method
num_tokens
(self)
llarp/policies/vis_bridge.py:80
Method
num_tokens
(self)
llarp/policies/vis_bridge.py:154
Function
on_save_ckpt_callback
Process the saved checkpoints here if desired.
llarp/run.py:31
Method
output_shape
(self)
llarp/policies/visual_encoders.py:25
Method
output_shape
(self)
llarp/policies/visual_encoders.py:89
Method
parameters
(self)
llarp/policies/llm_policy.py:85
Method
pddl_problem
(self)
llarp/task/predicate_task.py:144
← previous
next →
301–400 of 462, ranked by callers