MCPcopy Create free account

hub / github.com/apple/ml-llarp / functions

Functions462 in github.com/apple/ml-llarp

⨍ Functions 462◇ Types & classes 125↳ Endpoints 3
↓ 18 callersMethod__init__
(self, *args, **kwargs)
llarp/task/measures.py:446
↓ 12 callersMethod_get_uuid
(*args, **kwargs)
llarp/task/measures.py:450
↓ 10 callersMethodupdate
( self, rollouts, )
llarp/trainer/transformer_ppo.py:286
↓ 9 callersMethodparameters
(self)
llarp/policies/hl_policy.py:196
↓ 9 callersMethodto
(self, device)
llarp/policies/hl_policy.py:206
↓ 8 callersMethodclose
(self)
llarp/trainer/test_env_factory.py:121
↓ 7 callersMethod__init__
(self, *args, config, task, **kwargs)
llarp/task/sensors.py:265
↓ 7 callersMethodclone
(self)
llarp/dataset/generator.py:78
↓ 7 callersMethodget_sampled
(self)
llarp/task/predicate_task.py:281
↓ 7 callersMethodinsert
( self, next_observations=None, next_recurrent_hidden_states=None, actions=Non
llarp/policies/transformer_storage.py:98
↓ 7 callersMethodload
(filepath)
llarp/dataset/demo_dataset.py:50
↓ 6 callersMethodforward
(self, vis_features, obs)
llarp/policies/vis_bridge.py:17
↓ 6 callersFunctionget_parser
(llm_id)
llarp/task/utils.py:125
↓ 5 callersFunctionget_allowed_actions
(pddl, allowed_substrings)
llarp/task/utils.py:27
↓ 5 callersMethodsample
(self, sample_shape=torch.Size())
llarp/policies/action_decoders.py:22
↓ 5 callersMethodto
(self, device)
llarp/policies/llama_parallel.py:23
↓ 5 callersMethodupdate_metric
(self, *args, **kwargs)
llarp/task/measures.py:456
↓ 4 callersMethod_get_preds
(self)
llarp/dataset/dataset_validator.py:188
↓ 4 callersFunctionget_category_info
Get the list of all categories and a mapping from object name to category.
llarp/dataset/utils.py:42
↓ 4 callersMethodget_distrib
( self, observations, rnn_hidden_states, masks, rnn_build_seq_info=Non
llarp/policies/hl_policy.py:253
↓ 4 callersFunctionget_obs
(env)
llarp/dataset/dataset_validator.py:133
↓ 4 callersMethodlog_probs
(self, actions)
llarp/policies/action_decoders.py:25
↓ 4 callersMethodparameters
(self)
llarp/policies/utils.py:155
↓ 4 callersMethodpost_step
(self, obs)
llarp/trainer/test_env_factory.py:85
↓ 4 callersMethodreset_metric
(self, *args, **kwargs)
llarp/task/measures.py:453
↓ 4 callersMethodstep
(self, *args, sel, task, **kwargs)
llarp/task/actions.py:160
↓ 4 callersFunctionsummarize_episodes
(episodes, show_examples=False, tokenizer_name=None)
llarp/dataset/create_episodes.py:59
↓ 4 callersMethodto
(self, device)
llarp/policies/llm_policy.py:94
↓ 4 callersMethodtrain
(self)
llarp/trainer/trainer_loop.py:139
↓ 3 callersFunction_extract_pred_name
(expr, i)
llarp/task/measures.py:498
↓ 3 callersMethoddecode
( self, non_causal_tokens, causal_embeds, att_masks, kv_cache=None,
llarp/policies/utils.py:30
↓ 3 callersMethodembed_action
(self, action, llm)
llarp/policies/action_decoders.py:117
↓ 3 callersFunctionextract_entities
(expr)
llarp/dataset/dataset_validator.py:106
↓ 3 callersFunctionget_instruct_data
()
llarp/dataset/utils.py:21
↓ 3 callersMethodis_goal_satisfied
(self)
llarp/task/predicate_task.py:211
↓ 3 callersMethodload_state_dict
(self, state_dict, **kwargs)
llarp/policies/hl_policy.py:95
↓ 3 callersFunctionpred_to_str
(pred)
llarp/dataset/dataset_validator.py:60
↓ 3 callersMethodreset
(self, *args, task, **kwargs)
llarp/task/actions.py:156
↓ 3 callersMethodto
(self, device)
llarp/policies/utils.py:264
↓ 3 callersMethodto
(self, device)
llarp/policies/cores/decoder.py:181
↓ 3 callersMethodto_binary
(self)
llarp/dataset/episodes.py:64
↓ 3 callersMethodtokenizer
(self)
llarp/task/predicate_task.py:68
↓ 2 callersMethod__init__
(self, *args, config, task, **kwargs)
llarp/task/actions.py:137
↓ 2 callersMethod__init__
(self, config, obs_space, action_space, **kwargs)
llarp/policies/llm_policy.py:39
↓ 2 callersFunction_advance_cache
( kv_cache, kv_cache_masks, kv_counts, n_cur_toks, n_next_toks, new_kv_cache, prom
llarp/policies/cores/decoder.py:598
↓ 2 callersMethod_compute_actions_and_step_envs
Modified over the base implementation to support the multi-discrete action space.
llarp/trainer/trainer_loop.py:498
↓ 2 callersMethod_get_cur_ep
(self)
llarp/dataset/generator.py:273
↓ 2 callersMethod_get_non_causal_mask
(self, seq_len)
llarp/policies/utils.py:269
↓ 2 callersMethod_get_pred_distance
(self, pred, sim_info)
llarp/task/measures.py:518
↓ 2 callersMethod_group_scenes
(self, episodes)
llarp/dataset/episodes.py:275
↓ 2 callersMethod_include_prompt
Operates over language token ids.
llarp/policies/cores/decoder.py:474
↓ 2 callersMethod_on_successful_sample
(self)
llarp/dataset/generator.py:371
↓ 2 callersMethod_parse_instruct_id
(self, instruct_id)
llarp/task/measures.py:484
↓ 2 callersMethod_print_stats
(self)
llarp/dataset/dataset_validator.py:392
↓ 2 callersFunction_process_instruct
(phrasing: str, new_entities, name_remap)
llarp/dataset/generator.py:378
↓ 2 callersMethod_set_shuffle_intervals
(self)
llarp/dataset/episodes.py:294
↓ 2 callersMethod_setup_core_preds
(self)
llarp/dataset/dataset_validator.py:222
↓ 2 callersFunction_shuffle_select_subset
( entity_set: Dict[str, List[str]], ratio_sel: float )
llarp/dataset/generator.py:507
↓ 2 callersMethod_update_from_batch
Performs a gradient update from the minibatch.
llarp/trainer/transformer_ppo.py:340
↓ 2 callersMethodact
( self, observations, rnn_hidden_states, prev_actions, masks,
llarp/policies/llm_policy.py:138
↓ 2 callersMethodafter_update
(self)
llarp/trainer/transformer_ppo.py:145
↓ 2 callersFunctioneasy_register
(gym_id, local_path, overrides=None)
llarp/task/__init__.py:16
↓ 2 callersMethodembed_tokens
(self, tokens)
llarp/policies/utils.py:210
↓ 2 callersMethodentropy
(self)
llarp/policies/action_decoders.py:37
↓ 2 callersFunctionfix_top_down_cam_pos
(sim)
llarp/task/predicate_task.py:346
↓ 2 callersMethodflush
(self)
llarp/dataset/demo_dataset.py:141
↓ 2 callersMethodforward
(self, x, obs)
llarp/policies/llm_policy.py:238
↓ 2 callersMethodforward_norm
(self, x, obs)
llarp/policies/llm_policy.py:241
↓ 2 callersMethodget_num_rnn_layers
(self)
llarp/policies/cores/base_core.py:43
↓ 2 callersFunctionget_pddl
(task_config, all_cats, obj_cats)
llarp/task/utils.py:100
↓ 2 callersFunctionget_pred_hash
Hashes to return an ID based only on RELEVANT predicates
llarp/dataset/dataset_validator.py:72
↓ 2 callersMethodget_trainable_params
(self)
llarp/policies/cores/base_core.py:47
↓ 2 callersFunctionis_multi_node
()
llarp/trainer/custom_ddp.py:31
↓ 2 callersMethodmode
(self)
llarp/policies/action_decoders.py:34
↓ 2 callersFunctionpost_config_setup
(config)
test/test_llarp.py:40
↓ 2 callersMethodpost_process_returns
(self, returns, obs)
llarp/policies/llm_policy.py:235
↓ 2 callersFunctionpre_config_setup
()
test/test_llarp.py:29
↓ 2 callersMethodreset
(self)
llarp/trainer/test_env_factory.py:52
↓ 2 callersMethodrollout_actions
(self, obs, masks, deterministic=False, only_features=False)
llarp/policies/cores/decoder.py:541
↓ 2 callersFunctionstack_obs
(obs: List[Dict[str, np.ndarray]])
llarp/dataset/dataset_validator.py:55
↓ 2 callersMethodvis_bridge
(self)
llarp/policies/cores/decoder.py:197
↓ 1 callersMethod__init__
(self, *args, action_space, **kwargs)
llarp/policies/hl_policy.py:156
↓ 1 callersMethod__init__
( self, vis_encoder_net, llm, state_obs_space, hidden_size, cf
llarp/policies/vis_bridge.py:89
↓ 1 callersFunction_add_text_at_bottom
(frame, txt)
llarp/trainer/trainer_loop.py:728
↓ 1 callersFunction_auto_regr_sample_actions_kv_cache
( n_action_toks: int, llm: LlmWrapper, action_decoder_net: ActionDecoder, only_features: bool,
llarp/policies/cores/decoder.py:704
↓ 1 callersMethod_coalesce_post_step
(self, losses, count_steps_delta: int)
llarp/trainer/trainer_loop.py:103
↓ 1 callersMethod_collect_environment_result
(self, buffer_index: int = 0)
llarp/trainer/trainer_loop.py:301
↓ 1 callersMethod_compute_subgoals
Performs BFS to find a path from the start to the predicate goal state. Returns the predicate goals and the number of bfs iterations
llarp/dataset/dataset_validator.py:265
↓ 1 callersMethod_compute_subgoals_from_sol
(self, env, ordered_actions, sols, instruct_id)
llarp/dataset/dataset_validator.py:407
↓ 1 callersMethod_compute_var_mean
(x)
llarp/trainer/transformer_ppo.py:371
↓ 1 callersMethod_construct_kv_cache
(self, obs, masks)
llarp/policies/cores/decoder.py:496
↓ 1 callersFunction_construct_lang_tok_kv_cache
:returns: Updated kv cache, kv masks, and kv counts.
llarp/policies/cores/decoder.py:650
↓ 1 callersFunction_create_kv_cache
( context_len, n_heads, embed_dim, n_layers, batch_size, device, dtype )
llarp/policies/cores/decoder.py:56
↓ 1 callersFunction_create_recep_sampler
( recep_sampler_name, restrict_recep, exclude_recep, sampled_entities, receptacle_sets, )
llarp/dataset/generator.py:425
↓ 1 callersMethod_evaluate_actions
(self, *args, **kwargs)
llarp/trainer/transformer_ppo.py:395
↓ 1 callersFunction_extract_from_kv_cache
Extracts from the RIGHT SIDE (left side is chopped off).
llarp/policies/cores/decoder.py:808
↓ 1 callersFunction_filter_matching_pred_conds
Filters elements from matches based on a set of PDDL predicate strings.
llarp/task/predicate_task.py:314
↓ 1 callersMethod_filter_relevant_actions
(self, actions)
llarp/dataset/dataset_validator.py:160
↓ 1 callersFunction_flatten_eps
(all_eps, num_eps, instruct_samples)
llarp/dataset/generator.py:740
↓ 1 callersMethod_flush_frames
Write the frames to wandb.
llarp/trainer/trainer_loop.py:267
next →1–100 of 462, ranked by callers