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
↓ 1 callersMethod_forced_scene_switch
(self)
llarp/dataset/episodes.py:253
↓ 1 callersMethod_forced_scene_switch_if
(self)
llarp/dataset/episodes.py:307
↓ 1 callersMethod_generate_ao_states
(self)
llarp/dataset/generator.py:195
↓ 1 callersMethod_generate_objects
(self, ep_scene_handle, cur_ep)
llarp/dataset/generator.py:213
↓ 1 callersFunction_get_all_instructs
Gets the set of all possible distinct instructions grouped per instruct worker.
llarp/dataset/generator.py:521
↓ 1 callersMethod_get_cur_actions
(self, node)
llarp/dataset/dataset_validator.py:247
↓ 1 callersMethod_get_hidden_window_eval
(self, obs, att_masks, actions)
llarp/policies/cores/decoder.py:234
↓ 1 callersMethod_get_hidden_window_rollout
(self, obs)
llarp/policies/cores/decoder.py:340
↓ 1 callersMethod_get_instruct_sols
(self)
llarp/dataset/dataset_validator.py:152
↓ 1 callersMethod_get_rl_loss
(self, batch, learner_metrics, epoch, rollouts)
llarp/trainer/transformer_ppo.py:153
↓ 1 callersFunction_get_sample_instructs
List of instruction IDs t sample. There will be repeats depending on the overweight_freq.
llarp/dataset/generator.py:601
↓ 1 callersMethod_get_subgoals
(self, episode)
llarp/task/predicate_task.py:220
↓ 1 callersMethod_init_train
(self, resume_state=None)
llarp/trainer/trainer_loop.py:606
↓ 1 callersFunction_is_semantic_match
(ep_dat, sem_info: SemanticInstructInfo)
llarp/dataset/generator.py:791
↓ 1 callersMethod_load_goal_preds
(self, episode)
llarp/task/predicate_task.py:169
↓ 1 callersMethod_load_sampled_names
(self)
llarp/task/predicate_task.py:97
↓ 1 callersMethod_load_start_goal
Setup the start and goal PDDL conditions. Will change the simulator state to set the start state.
llarp/task/predicate_task.py:72
↓ 1 callersMethod_load_start_preds
(self, episode)
llarp/task/predicate_task.py:182
↓ 1 callersFunction_map_state_dict
(state_dict, rename_ckpt_keys)
llarp/policies/hl_policy.py:33
↓ 1 callersFunction_map_state_dict_k
(state_dict_k, rename_ckpt_keys)
llarp/policies/hl_policy.py:40
↓ 1 callersFunction_parse_instruct
( instruct: Dict[str, Any], instruct_id: str, valid_obj_cls_names: List[str] )
llarp/dataset/generator.py:688
↓ 1 callersMethod_post_act
(self, actions, hidden_window, obs)
llarp/policies/cores/decoder.py:325
↓ 1 callersMethod_randomize_value
(value: int, value_range: float)
llarp/dataset/episodes.py:289
↓ 1 callersFunction_recur_dict_replace
Replace all string entries in `d` with the replace name to PDDL entity mapping in replaces.
llarp/dataset/generator.py:671
↓ 1 callersFunction_recur_dict_replace
Replace all string entries in `d` with the replace name to PDDL entity mapping in replaces.
llarp/task/predicate_task.py:291
↓ 1 callersMethod_rollout
(self, count_steps_delta)
llarp/trainer/trainer_loop.py:587
↓ 1 callersFunction_sample_objects
( obj_sample_split, sampled_objs: List[Tuple[str, ObjectSampleInfo]], ep_dats: List[EpisodeData],
llarp/dataset/generator.py:447
↓ 1 callersFunction_sample_receps
( sampled_receps: List[Tuple[str, ReceptacleSampleInfo]], ep_dats: List[EpisodeData], receptacle_s
llarp/dataset/generator.py:397
↓ 1 callersMethod_setup_pddl_entities
(self, episode)
llarp/task/predicate_task.py:155
↓ 1 callersMethod_should_save_resume_state
(self)
llarp/trainer/trainer_loop.py:73
↓ 1 callersMethod_shuffle
(self)
llarp/dataset/episodes.py:264
↓ 1 callersFunction_strip_instance_id
(instance_id: str)
llarp/task/predicate_task.py:285
↓ 1 callersFunction_tokenize_prompts
( prompts: Dict[str, str], prompt_suffix: str, tokenizer )
llarp/policies/cores/decoder.py:44
↓ 1 callersMethod_training_log
(self, writer, losses: dict[str, float], prev_time: int = 0)
llarp/trainer/trainer_loop.py:427
↓ 1 callersMethod_update_agent
(self)
llarp/trainer/trainer_loop.py:558
↓ 1 callersMethod_verify_input
(self, non_causal_tokens, causal_embeds, att_masks)
llarp/policies/utils.py:183
↓ 1 callersFunction_write_to_cache
Writes to the end of the seq dim.
llarp/policies/cores/decoder.py:633
↓ 1 callersMethodapply_ee_constraints
(self)
llarp/task/actions.py:93
↓ 1 callersMethodasync_step_at
(self, index_env, action)
llarp/trainer/test_env_factory.py:88
↓ 1 callersFunctioncheck_hidden_window
(hidden_window, att_masks)
llarp/policies/cores/decoder.py:823
↓ 1 callersMethodcollect_action
(self, action, infos)
llarp/dataset/demo_dataset.py:93
↓ 1 callersMethodcollect_obs
Add observation to all workers.
llarp/dataset/demo_dataset.py:105
↓ 1 callersFunctioncompact_str_no_robot
(action: PddlAction)
llarp/dataset/dataset_validator.py:623
↓ 1 callersMethodcompute_returns
(self, next_value, use_gae, gamma, tau, clip_rewards: int = 0.0)
llarp/policies/transformer_storage.py:170
↓ 1 callersFunctionconvert_rearrange_ep_to_lang_rearrange
( ep: RearrangeEpisode, )
llarp/task/predicate_task.py:365
↓ 1 callersMethoddata_generator
( self, advantages: Optional[torch.Tensor], num_mini_batch: int, norm_returns,
llarp/policies/transformer_storage.py:203
↓ 1 callersMethoddisplay_model_info
(self)
llarp/policies/cores/decoder.py:163
↓ 1 callersFunctionextract_scalars_from_infos_safe
( infos: list[dict[str, Any]], ignore_keys: Optional[set[str]], n_envs: int, device, )
llarp/trainer/trainer_loop.py:52
↓ 1 callersFunctionflatten_actions
(pddl: PddlDomain, obj_cats)
llarp/task/utils.py:47
↓ 1 callersFunctionflatten_trajs
( advantages, buffers, context_len: int, current_rollout_step_idx, hidden_window, befo
llarp/policies/transformer_storage.py:235
↓ 1 callersMethodforward
(self, x, obs)
llarp/policies/llm_policy.py:198
↓ 1 callersMethodfrom_binary
( self, data_dict: Dict[str, Any], scenes_dir: Optional[str] = None )
llarp/dataset/episodes.py:131
↓ 1 callersMethodfrom_config
(cls, config, observation_space, action_space, **kwargs)
llarp/policies/llm_policy.py:172
↓ 1 callersFunctiongenerate_all_instructions
( instruct_path: str, obj_sets: Dict[str, List[str]], recep_sets: Dict[str, List[str]], instru
llarp/dataset/generator.py:618
↓ 1 callersMethodgenerate_episodes
(self, num_episodes: int = 1, verbose: bool = False)
llarp/dataset/generator.py:127
↓ 1 callersMethodgenerate_scene
Gets the scene ID for the current episode. Is fixed by the process ID.
llarp/dataset/generator.py:113
↓ 1 callersMethodgenerate_single_episode
(self)
llarp/dataset/generator.py:276
↓ 1 callersMethodget_advantages
(self, rollouts)
llarp/trainer/transformer_ppo.py:274
↓ 1 callersFunctionget_all_instruct_ids
()
llarp/dataset/utils.py:37
↓ 1 callersFunctionget_debug
(hidden_window)
llarp/policies/cores/decoder.py:846
↓ 1 callersMethodget_extra
(self, action_data, infos, dones)
llarp/policies/hl_policy.py:117
↓ 1 callersFunctionget_flat_eps_split
Deterministically splits the episodes and selects the episode split for `proc_idx`.
llarp/dataset/generator.py:756
↓ 1 callersFunctionget_main_addr
()
llarp/trainer/custom_ddp.py:35
↓ 1 callersFunctionget_name_mappings
Gets the friendly name mappings from the instruction file.
llarp/dataset/utils.py:29
↓ 1 callersFunctionget_obj_type
(cat, pddl)
llarp/task/utils.py:96
↓ 1 callersMethodget_observation
(self, *args, **kwargs)
llarp/task/sensors.py:295
↓ 1 callersMethodget_value
(self, observations, rnn_hidden_states, prev_actions, masks)
llarp/policies/hl_policy.py:217
↓ 1 callersFunctionhabitat_create_dataset
( config, is_first_rank, num_environments, force_scene_per_worker: bool, filter_down_num:
llarp/trainer/custom_env_factory.py:36
↓ 1 callersFunctioninit_distrib_slurm
( backend: str = "nccl", )
llarp/trainer/custom_ddp.py:43
↓ 1 callersMethodinit_distributed
(self, find_unused_params: bool = True)
llarp/trainer/transformer_ppo.py:374
↓ 1 callersMethodinsert_first_observations
(self, batch)
llarp/policies/transformer_storage.py:145
↓ 1 callersFunctionmain
(cfg)
llarp/run.py:45
↓ 1 callersFunctionmatches_any
(s, allowed_acs)
llarp/task/utils.py:30
↓ 1 callersMethodon_ep_done
(self, env_i, env_infos, instruction: Optional[str] = None)
llarp/dataset/demo_dataset.py:114
↓ 1 callersMethodsave
(self, filepath)
llarp/dataset/demo_dataset.py:42
↓ 1 callersMethodset_desired_ee_pos
(self, ee_delta: np.ndarray)
llarp/task/actions.py:100
↓ 1 callersFunctionsetup_peft_module
(model, peft_full_att_params, peft_settings)
llarp/policies/utils.py:295
↓ 1 callersMethodshould_checkpoint
(self)
llarp/trainer/trainer_loop.py:88
↓ 1 callersFunctionsplit_scenes
(scenes, n_envs)
llarp/trainer/custom_env_factory.py:21
↓ 1 callersFunctionstart
(args)
llarp/dataset/create_episodes.py:118
↓ 1 callersFunctionstart
(args)
llarp/dataset/dataset_validator.py:550
↓ 1 callersFunctionstart
(args)
llarp/dataset/combine_datasets.py:42
↓ 1 callersFunctionsubsel_llm_hidden_out
( llm_hidden_out, real_n_steps, debug_mode, fetch_befores, is_real_token, att_masks )
llarp/policies/cores/decoder.py:850
↓ 1 callersMethodto
(self, device)
llarp/policies/transformer_storage.py:92
↓ 1 callersFunctiontranspose_stack_pad_dicts
(dicts_i)
llarp/policies/transformer_storage.py:32
↓ 1 callersMethodupdate_actions
(self, obs, actions)
llarp/policies/cores/decoder.py:366
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:33
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:52
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:71
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:91
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:117
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:136
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:156
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:183
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:225
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:275
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:317
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:346
↓ 1 callersMethodupdate_metric
(self, *args, task, **kwargs)
llarp/task/measures.py:393
↓ 1 callersMethodupdate_metric
(self, *args, **kwargs)
llarp/task/measures.py:440
← previousnext →101–200 of 462, ranked by callers