SYMBOL INDEX (2386 symbols across 212 files) FILE: allenact/algorithms/offpolicy_sync/losses/abstract_offpolicy_loss.py class AbstractOffPolicyLoss (line 14) | class AbstractOffPolicyLoss(Generic[ModelType], Loss): method loss (line 20) | def loss( # type: ignore FILE: allenact/algorithms/onpolicy_sync/engine.py class OnPolicyRLEngine (line 93) | class OnPolicyRLEngine(object): method __init__ (line 103) | def __init__( method vector_tasks (line 302) | def vector_tasks( method worker_seeds (line 344) | def worker_seeds(nprocesses: int, initial_seed: Optional[int]) -> List... method get_sampler_fn_args (line 356) | def get_sampler_fn_args(self, seeds: Optional[List[int]] = None): method checkpoint_load (line 397) | def checkpoint_load( method aggregate_task_metrics (line 422) | def aggregate_task_metrics( method _preprocess_observations (line 457) | def _preprocess_observations(self, batched_observations): method remove_paused (line 462) | def remove_paused(self, observations): method initialize_storage_and_viz (line 479) | def initialize_storage_and_viz( method num_active_samplers (line 527) | def num_active_samplers(self): method act (line 532) | def act( method aggregate_and_send_logging_package (line 549) | def aggregate_and_send_logging_package( method _active_memory (line 598) | def _active_memory(memory, keep): method probe (line 601) | def probe(self, dones: List[bool], npaused, period=100000): method collect_step_across_all_task_samplers (line 656) | def collect_step_across_all_task_samplers( method distributed_weighted_sum (line 776) | def distributed_weighted_sum( method distributed_reduce (line 793) | def distributed_reduce( method backprop_step (line 804) | def backprop_step( method save_error_data (line 812) | def save_error_data(self, batch: Dict[str, Any]): method step_count (line 816) | def step_count(self) -> int: method compute_losses_track_them_and_backprop (line 823) | def compute_losses_track_them_and_backprop( method close (line 1108) | def close(self, verbose=True): method is_closed (line 1140) | def is_closed(self): method is_closing (line 1144) | def is_closing(self): method __del__ (line 1147) | def __del__(self): method __enter__ (line 1150) | def __enter__(self): method __exit__ (line 1153) | def __exit__(self, exc_type, exc_val, exc_tb): class OnPolicyTrainer (line 1157) | class OnPolicyTrainer(OnPolicyRLEngine): method __init__ (line 1158) | def __init__( method advance_seed (line 1272) | def advance_seed( method deterministic_seeds (line 1290) | def deterministic_seeds(self) -> None: method save_error_data (line 1299) | def save_error_data(self, batch: Dict[str, Any]) -> str: method aggregate_and_send_logging_package (line 1329) | def aggregate_and_send_logging_package( method checkpoint_save (line 1353) | def checkpoint_save(self, pipeline_stage_index: Optional[int] = None) ... method checkpoint_load (line 1383) | def checkpoint_load( method step_count (line 1405) | def step_count(self): method step_count (line 1409) | def step_count(self, val: int) -> None: method log_interval (line 1413) | def log_interval(self): method approx_steps (line 1419) | def approx_steps(self): method act (line 1428) | def act( method advantage_stats (line 1465) | def advantage_stats(self, advantages: torch.Tensor) -> Dict[str, torch... method backprop_step (line 1495) | def backprop_step( method _save_checkpoint_then_send_checkpoint_for_validation_and_update_last_save_counter (line 1538) | def _save_checkpoint_then_send_checkpoint_for_validation_and_update_la... method run_pipeline (line 1553) | def run_pipeline(self, valid_on_initial_weights: bool = False): method train (line 1897) | def train( class OnPolicyInference (line 1937) | class OnPolicyInference(OnPolicyRLEngine): method __init__ (line 1938) | def __init__( method run_eval (line 1976) | def run_eval( method skip_to_latest (line 2321) | def skip_to_latest(checkpoints_queue: mp.Queue, command: Optional[str]... method process_checkpoints (line 2354) | def process_checkpoints(self): FILE: allenact/algorithms/onpolicy_sync/losses/a2cacktr.py class A2CACKTR (line 16) | class A2CACKTR(AbstractActorCriticLoss): method __init__ (line 28) | def __init__( method loss_per_step (line 49) | def loss_per_step( # type: ignore method loss (line 94) | def loss( # type: ignore class A2C (line 129) | class A2C(A2CACKTR): method __init__ (line 132) | def __init__( class ACKTR (line 150) | class ACKTR(A2CACKTR): method __init__ (line 157) | def __init__( FILE: allenact/algorithms/onpolicy_sync/losses/abstract_loss.py class AbstractActorCriticLoss (line 13) | class AbstractActorCriticLoss(Loss): method loss (line 19) | def loss( # type: ignore FILE: allenact/algorithms/onpolicy_sync/losses/grouped_action_imitation.py class GroupedActionImitation (line 14) | class GroupedActionImitation(AbstractActorCriticLoss): method __init__ (line 15) | def __init__( method loss (line 34) | def loss( # type: ignore FILE: allenact/algorithms/onpolicy_sync/losses/imitation.py class Imitation (line 23) | class Imitation(AbstractActorCriticLoss): method __init__ (line 26) | def __init__( method group_loss (line 34) | def group_loss( method loss (line 66) | def loss( # type: ignore FILE: allenact/algorithms/onpolicy_sync/losses/ppo.py class PPO (line 15) | class PPO(AbstractActorCriticLoss): method __init__ (line 32) | def __init__( method loss_per_step (line 62) | def loss_per_step( method loss (line 131) | def loss( # type: ignore class PPOValue (line 166) | class PPOValue(AbstractActorCriticLoss): method __init__ (line 175) | def __init__( method loss (line 192) | def loss( # type: ignore FILE: allenact/algorithms/onpolicy_sync/misc.py class TrackingInfoType (line 7) | class TrackingInfoType(Enum): class TrackingInfo (line 14) | class TrackingInfo: FILE: allenact/algorithms/onpolicy_sync/policy.py class ActorCriticModel (line 29) | class ActorCriticModel(Generic[DistributionType], nn.Module): method __init__ (line 40) | def __init__(self, action_space: gym.Space, observation_space: SpaceDi... method recurrent_memory_specification (line 54) | def recurrent_memory_specification(self) -> Optional[FullMemorySpecType]: method _recurrent_memory_specification (line 83) | def _recurrent_memory_specification(self) -> Optional[FullMemorySpecTy... method forward (line 108) | def forward( # type:ignore class LinearActorCriticHead (line 139) | class LinearActorCriticHead(nn.Module): method __init__ (line 140) | def __init__(self, input_size: int, num_actions: int): method forward (line 149) | def forward(self, x) -> Tuple[CategoricalDistr, torch.Tensor]: class LinearCriticHead (line 163) | class LinearCriticHead(nn.Module): method __init__ (line 164) | def __init__(self, input_size: int): method forward (line 170) | def forward(self, x): class LinearActorHead (line 174) | class LinearActorHead(nn.Module): method __init__ (line 175) | def __init__(self, num_inputs: int, num_outputs: int): method forward (line 182) | def forward(self, x: torch.FloatTensor): # type: ignore FILE: allenact/algorithms/onpolicy_sync/runner.py class SaveDirFormat (line 63) | class SaveDirFormat(enum.Enum): class OnPolicyRunner (line 80) | class OnPolicyRunner(object): method __init__ (line 81) | def __init__( method callbacks (line 153) | def callbacks(self): method local_start_time_str (line 157) | def local_start_time_str(self) -> str: method running_validation (line 166) | def running_validation(self): method init_context (line 182) | def init_context( method setup_callback_classes (line 202) | def setup_callback_classes(self, callbacks: Optional[str]) -> Set[Call... method _acquire_unique_local_start_time_string (line 261) | def _acquire_unique_local_start_time_string(self) -> str: method worker_devices (line 309) | def worker_devices(self, mode: str): method local_worker_ids (line 322) | def local_worker_ids(self, mode: str): method init_visualizer (line 334) | def init_visualizer(self, mode: str): method init_process (line 343) | def init_process(mode: str, id: int, to_close_on_termination: OnPolicy... method init_worker (line 380) | def init_worker(engine_class, args, kwargs): method _get_callback_sensors (line 395) | def _get_callback_sensors(self) -> List[Sensor]: method train_loop (line 404) | def train_loop( method valid_loop (line 432) | def valid_loop(id: int = 0, *engine_args, **engine_kwargs): method test_loop (line 445) | def test_loop(id: int = 0, *engine_args, **engine_kwargs): method _initialize_start_train_or_start_test (line 455) | def _initialize_start_train_or_start_test(self): method get_port (line 477) | def get_port(self): method start_train (line 495) | def start_train( method start_test (line 671) | def start_test( method checkpoint_start_time_str (line 772) | def checkpoint_start_time_str(checkpoint_file_name): method checkpoint_log_folder_str (line 780) | def checkpoint_log_folder_str(checkpoint_file_name): method experiment_name (line 788) | def experiment_name(self): method checkpoint_dir (line 793) | def checkpoint_dir( method log_writer_path (line 822) | def log_writer_path(self, start_time_str: str) -> str: method metric_path (line 859) | def metric_path(self, start_time_str: str) -> str: method save_project_state (line 881) | def save_project_state(self): method _update_keys (line 961) | def _update_keys( method _process_logging_packages (line 980) | def _process_logging_packages( method process_valid_package (line 1261) | def process_valid_package( method process_train_packages (line 1276) | def process_train_packages( method process_test_packages (line 1292) | def process_test_packages( method log_and_close (line 1307) | def log_and_close( method get_checkpoint_files (line 1509) | def get_checkpoint_files( method step_from_checkpoint (line 1557) | def step_from_checkpoint(ckpt_path: str) -> int: method close (line 1574) | def close(self, verbose=True): method __del__ (line 1608) | def __del__(self): method __enter__ (line 1611) | def __enter__(self): method __exit__ (line 1614) | def __exit__(self, exc_type, exc_val, exc_tb): FILE: allenact/algorithms/onpolicy_sync/storage.py class ExperienceStorage (line 34) | class ExperienceStorage(abc.ABC): method initialize (line 36) | def initialize(self, *, observations: ObservationType, **kwargs): method add (line 40) | def add( method before_updates (line 65) | def before_updates(self, **kwargs): method after_updates (line 68) | def after_updates(self, **kwargs) -> int: method to (line 72) | def to(self, device: torch.device): method set_partition (line 76) | def set_partition(self, index: int, num_parts: int): method total_experiences (line 81) | def total_experiences(self) -> int: class RolloutStorage (line 85) | class RolloutStorage(ExperienceStorage, abc.ABC): method initialize (line 88) | def initialize( method agent_input_for_next_step (line 100) | def agent_input_for_next_step(self) -> Dict[str, Any]: method sampler_select (line 104) | def sampler_select(self, keep_list: Sequence[int]): class StreamingStorageMixin (line 108) | class StreamingStorageMixin(abc.ABC): method next_batch (line 110) | def next_batch(self) -> Dict[str, Any]: method reset_stream (line 113) | def reset_stream(self): method empty (line 117) | def empty(self) -> bool: class MiniBatchStorageMixin (line 121) | class MiniBatchStorageMixin(abc.ABC): method batched_experience_generator (line 123) | def batched_experience_generator( class RolloutBlockStorage (line 130) | class RolloutBlockStorage(RolloutStorage, MiniBatchStorageMixin): method __init__ (line 135) | def __init__(self, init_size: int = 50): method initialize (line 173) | def initialize( method total_experiences (line 211) | def total_experiences(self) -> int: method total_experiences (line 215) | def total_experiences(self, value: int): method set_partition (line 218) | def set_partition(self, index: int, num_parts: int): method value_preds (line 222) | def value_preds(self) -> torch.Tensor: method rewards (line 226) | def rewards(self) -> torch.Tensor: method returns (line 230) | def returns(self) -> torch.Tensor: method action_log_probs (line 234) | def action_log_probs(self) -> torch.Tensor: method actions (line 238) | def actions(self) -> torch.Tensor: method prev_actions (line 242) | def prev_actions(self) -> torch.Tensor: method masks (line 246) | def masks(self) -> torch.Tensor: method observations (line 250) | def observations(self) -> Memory: method create_memory (line 254) | def create_memory( method to (line 279) | def to(self, device: torch.device): method insert_observations (line 297) | def insert_observations( method insert_memory (line 309) | def insert_memory( method insert_tensors (line 329) | def insert_tensors( method create_tensor_storage (line 400) | def create_tensor_storage( method _double_storage_size (line 405) | def _double_storage_size(self): method add (line 438) | def add( method sampler_select (line 502) | def sampler_select(self, keep_list: Sequence[int]): method before_updates (line 519) | def before_updates( method after_updates (line 546) | def after_updates(self, **kwargs): method _extend_tensor_with_ones (line 565) | def _extend_tensor_with_ones(stored_tensor: torch.Tensor, desired_num_... method compute_returns (line 572) | def compute_returns( method batched_experience_generator (line 601) | def batched_experience_generator( method unflatten_observations (line 677) | def unflatten_observations(self, flattened_batch: Memory) -> Observati... method pick_observation_step (line 689) | def pick_observation_step(self, step: int) -> ObservationType: method pick_memory_step (line 692) | def pick_memory_step(self, step: int) -> Memory: method pick_prev_actions_step (line 696) | def pick_prev_actions_step(self, step: int) -> ActionType: method agent_input_for_next_step (line 699) | def agent_input_for_next_step(self) -> Dict[str, Any]: FILE: allenact/algorithms/onpolicy_sync/vector_sampled_tasks.py class DelaySignalHandling (line 68) | class DelaySignalHandling: method __init__ (line 70) | def __init__(self): method __enter__ (line 76) | def __enter__(self): method int_handler (line 82) | def int_handler(self, sig, frame): method term_handler (line 86) | def term_handler(self, sig, frame): method __exit__ (line 90) | def __exit__(self, type, value, traceback): class VectorSampledTasks (line 116) | class VectorSampledTasks: method __init__ (line 151) | def __init__( method _create_read_function_with_timeout (line 262) | def _create_read_function_with_timeout( method _reset_sampler_index_to_process_ind_and_subprocess_ind (line 280) | def _reset_sampler_index_to_process_ind_and_subprocess_ind(self): method _partition_to_processes (line 289) | def _partition_to_processes(self, seq: Union[Iterator, Sequence]): method is_closed (line 304) | def is_closed(self) -> bool: method num_unpaused_tasks (line 309) | def num_unpaused_tasks(self) -> int: method mp_ctx (line 319) | def mp_ctx(self): method _task_sampling_loop_worker (line 329) | def _task_sampling_loop_worker( method _spawn_workers (line 429) | def _spawn_workers( method next_task (line 483) | def next_task(self, **kwargs): method get_observations (line 498) | def get_observations(self): method command_at (line 509) | def command_at( method call_at (line 531) | def call_at( method next_task_at (line 556) | def next_task_at(self, sampler_index: int) -> List[RLStepResult]: method step_at (line 574) | def step_at(self, sampler_index: int, action: Any) -> List[RLStepResult]: method async_step (line 592) | def async_step(self, actions: Sequence[Any]) -> None: method wait_step (line 605) | def wait_step(self) -> List[Dict[str, Any]]: method step (line 613) | def step(self, actions: Sequence[Any]): method reset_all (line 627) | def reset_all(self): method set_seeds (line 632) | def set_seeds(self, seeds: List[int]): method close (line 641) | def close(self) -> None: method pause_at (line 671) | def pause_at(self, sampler_index: int) -> None: method resume_all (line 708) | def resume_all(self) -> None: method command (line 724) | def command( method call (line 748) | def call( method attr_at (line 786) | def attr_at(self, sampler_index: int, attr_name: str) -> Any: method attr (line 801) | def attr(self, attr_names: Union[List[str], str]) -> List[Any]: method render (line 817) | def render( method _valid_start_methods (line 844) | def _valid_start_methods(self) -> Set[str]: method __del__ (line 847) | def __del__(self): method __enter__ (line 850) | def __enter__(self): method __exit__ (line 853) | def __exit__(self, exc_type, exc_val, exc_tb): class SingleProcessVectorSampledTasks (line 857) | class SingleProcessVectorSampledTasks(object): method __init__ (line 881) | def __init__( method is_closed (line 936) | def is_closed(self) -> bool: method mp_ctx (line 941) | def mp_ctx(self) -> Optional[BaseContext]: method num_unpaused_tasks (line 945) | def num_unpaused_tasks(self) -> int: method _task_sampling_loop_generator_fn (line 955) | def _task_sampling_loop_generator_fn( method _create_generators (line 1096) | def _create_generators( method next_task (line 1125) | def next_task(self, **kwargs): method get_observations (line 1140) | def get_observations(self): method next_task_at (line 1151) | def next_task_at(self, index_process: int) -> List[RLStepResult]: method step_at (line 1167) | def step_at(self, index_process: int, action: int) -> List[RLStepResult]: method step (line 1181) | def step(self, actions: List[List[int]]): method reset_all (line 1197) | def reset_all(self): method set_seeds (line 1202) | def set_seeds(self, seeds: List[int]): method close (line 1214) | def close(self) -> None: method pause_at (line 1229) | def pause_at(self, sampler_index: int) -> None: method resume_all (line 1243) | def resume_all(self) -> None: method command_at (line 1249) | def command_at( method command (line 1267) | def command( method call_at (line 1284) | def call_at( method call (line 1307) | def call( method attr_at (line 1340) | def attr_at(self, sampler_index: int, attr_name: str) -> Any: method attr (line 1357) | def attr(self, attr_names: Union[List[str], str]) -> List[Any]: method render (line 1376) | def render( method __del__ (line 1405) | def __del__(self): method __enter__ (line 1408) | def __enter__(self): method __exit__ (line 1411) | def __exit__(self, exc_type, exc_val, exc_tb): FILE: allenact/base_abstractions/callbacks.py class Callback (line 13) | class Callback(abc.ABC): method setup (line 14) | def setup( method on_train_log (line 23) | def on_train_log( method on_valid_log (line 36) | def on_valid_log( method on_test_log (line 49) | def on_test_log( method after_save_project_state (line 62) | def after_save_project_state(self, base_dir: str) -> None: method callback_sensors (line 65) | def callback_sensors(self) -> Optional[Sequence[Sensor]]: FILE: allenact/base_abstractions/distributions.py class Distr (line 22) | class Distr(abc.ABC): method log_prob (line 24) | def log_prob(self, actions: Any): method entropy (line 29) | def entropy(self): method sample (line 34) | def sample(self, sample_shape=torch.Size()): method mode (line 38) | def mode(self): class CategoricalDistr (line 46) | class CategoricalDistr(torch.distributions.Categorical, Distr): method mode (line 53) | def mode(self): method log_prob (line 56) | def log_prob(self, value: torch.Tensor): method log_probs_tensor (line 75) | def log_probs_tensor(self): method probs_tensor (line 79) | def probs_tensor(self): class ConditionalDistr (line 83) | class ConditionalDistr(Distr): method __init__ (line 93) | def __init__( method log_prob (line 124) | def log_prob(self, actions): method entropy (line 127) | def entropy(self): method condition_on_input (line 130) | def condition_on_input(self, **ready_actions): method reset (line 142) | def reset(self): method sample (line 148) | def sample(self, sample_shape=torch.Size()) -> OrderedDict: method mode (line 151) | def mode(self) -> OrderedDict: class SequentialDistr (line 155) | class SequentialDistr(Distr): method __init__ (line 156) | def __init__(self, *conditional_distrs: ConditionalDistr): method sample (line 163) | def sample(self, sample_shape=torch.Size()): method mode (line 170) | def mode(self): method conditional_entropy (line 177) | def conditional_entropy(self): method entropy (line 183) | def entropy(self): method log_prob (line 189) | def log_prob( class TrackingCallback (line 212) | class TrackingCallback(Protocol): method __call__ (line 213) | def __call__(self, type: TrackingInfoType, info: Dict[str, Any], n: in... class TeacherForcingDistr (line 216) | class TeacherForcingDistr(Distr): method __init__ (line 217) | def __init__( method enforce (line 248) | def enforce( method log_prob (line 302) | def log_prob(self, actions: Any): method entropy (line 305) | def entropy(self): method conditional_entropy (line 308) | def conditional_entropy(self): method sample (line 316) | def sample(self, sample_shape=torch.Size()): class AddBias (line 353) | class AddBias(nn.Module): method __init__ (line 356) | def __init__(self, bias: torch.FloatTensor): method forward (line 366) | def forward(self, x: torch.FloatTensor) -> torch.FloatTensor: # type:... FILE: allenact/base_abstractions/experiment_config.py function split_processes_onto_devices (line 17) | def split_processes_onto_devices(nprocesses: int, ndevices: int): class MachineParams (line 27) | class MachineParams(object): method __init__ (line 28) | def __init__( method set_local_worker_ids (line 77) | def set_local_worker_ids(self, local_worker_ids: Optional[List[int]]): method instance_from (line 86) | def instance_from( method _standardize_devices (line 95) | def _standardize_devices( method sensor_preprocessor_graph (line 127) | def sensor_preprocessor_graph(self) -> Optional[SensorPreprocessorGraph]: method set_visualizer (line 143) | def set_visualizer(self, viz: VizSuite): method visualizer (line 150) | def visualizer(self) -> Optional[VizSuite]: class FrozenClassVariables (line 163) | class FrozenClassVariables(abc.ABCMeta): method __setattr__ (line 171) | def __setattr__(cls, attr, value): class ExperimentConfig (line 188) | class ExperimentConfig(metaclass=FrozenClassVariables): method tag (line 199) | def tag(self) -> str: method training_pipeline (line 204) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method machine_params (line 218) | def machine_params( method create_model (line 240) | def create_model(self, **kwargs) -> nn.Module: method make_sampler_fn (line 245) | def make_sampler_fn(self, **kwargs) -> TaskSampler: method train_task_sampler_args (line 256) | def train_task_sampler_args( method valid_task_sampler_args (line 284) | def valid_task_sampler_args( method test_task_sampler_args (line 300) | def test_task_sampler_args( FILE: allenact/base_abstractions/misc.py class RLStepResult (line 23) | class RLStepResult(NamedTuple): method clone (line 29) | def clone(self, new_info: Dict[str, Any]): method merge (line 41) | def merge(self, other: "RLStepResult"): class ActorCriticOutput (line 55) | class ActorCriticOutput(tuple, Generic[DistributionType]): method __new__ (line 61) | def __new__( method __repr__ (line 73) | def __repr__(self) -> str: class Memory (line 81) | class Memory(Dict): method __init__ (line 82) | def __init__(self, *args, **kwargs): method check_append (line 111) | def check_append( method tensor (line 146) | def tensor(self, key: str) -> torch.Tensor: method sampler_dim (line 160) | def sampler_dim(self, key: str) -> int: method sampler_select (line 174) | def sampler_select(self, keep: Sequence[int]) -> "Memory": method set_tensor (line 209) | def set_tensor(self, key: str, tensor: torch.Tensor) -> "Memory": method step_select (line 231) | def step_select(self, step: int) -> "Memory": method step_squeeze (line 261) | def step_squeeze(self, step: int) -> "Memory": method slice (line 286) | def slice( method to (line 346) | def to(self, device: torch.device) -> "Memory": class Loss (line 354) | class Loss(abc.ABC): class LossOutput (line 359) | class LossOutput: class GenericAbstractLoss (line 368) | class GenericAbstractLoss(Loss): method loss (line 371) | def loss( # type: ignore FILE: allenact/base_abstractions/preprocessor.py class Preprocessor (line 14) | class Preprocessor(abc.ABC): method __init__ (line 30) | def __init__( method process (line 42) | def process(self, obs: Dict[str, Any], *args: Any, **kwargs: Any) -> Any: method to (line 56) | def to(self, device: torch.device) -> "Preprocessor": class SensorPreprocessorGraph (line 60) | class SensorPreprocessorGraph: method __init__ (line 87) | def __init__( method get (line 147) | def get(self, uuid: str) -> Preprocessor: method to (line 160) | def to(self, device: torch.device) -> "SensorPreprocessorGraph": method get_observations (line 166) | def get_observations( class PreprocessorGraph (line 183) | class PreprocessorGraph(SensorPreprocessorGraph): method __init__ (line 184) | def __init__(self, *args, **kwargs): class ObservationSet (line 191) | class ObservationSet: method __init__ (line 192) | def __init__(self, *args, **kwargs) -> None: FILE: allenact/base_abstractions/sensor.py class Sensor (line 38) | class Sensor(Generic[EnvType, SubTaskType]): method __init__ (line 53) | def __init__(self, uuid: str, observation_space: gym.Space, **kwargs: ... method get_observation (line 57) | def get_observation( class SensorSuite (line 74) | class SensorSuite(Generic[EnvType]): method __init__ (line 87) | def __init__(self, sensors: Sequence[Sensor]) -> None: method get (line 104) | def get(self, uuid: str) -> Sensor: method get_observations (line 117) | def get_observations( class AbstractExpertSensor (line 137) | class AbstractExpertSensor(Sensor[EnvType, SubTaskType], abc.ABC): method __init__ (line 145) | def __init__( method flagged_group_space (line 210) | def flagged_group_space(cls, group_space: gym.spaces.Space) -> gym.spa... method flagged_space (line 222) | def flagged_space( method _get_observation_space (line 248) | def _get_observation_space(self) -> gym.spaces.Dict: method _zeroed_observation (line 261) | def _zeroed_observation(self) -> Union[OrderedDict, Tuple]: method flatten_output (line 271) | def flatten_output(self, unflattened): method query_expert (line 282) | def query_expert( method get_observation (line 299) | def get_observation( class AbstractExpertActionSensor (line 327) | class AbstractExpertActionSensor(AbstractExpertSensor, abc.ABC): method __init__ (line 328) | def __init__( method flagged_group_space (line 340) | def flagged_group_space(cls, group_space: gym.spaces.Space) -> gym.spa... class ExpertActionSensor (line 356) | class ExpertActionSensor(AbstractExpertActionSensor): method query_expert (line 360) | def query_expert( class AbstractExpertPolicySensor (line 368) | class AbstractExpertPolicySensor(AbstractExpertSensor, abc.ABC): method __init__ (line 369) | def __init__( method flagged_group_space (line 381) | def flagged_group_space(cls, group_space: gym.spaces.Space) -> gym.spa... class ExpertPolicySensor (line 398) | class ExpertPolicySensor(AbstractExpertPolicySensor): method query_expert (line 402) | def query_expert( FILE: allenact/base_abstractions/task.py class Task (line 23) | class Task(Generic[EnvType]): method __init__ (line 48) | def __init__( method get_observations (line 66) | def get_observations(self, **kwargs) -> Any: method action_space (line 71) | def action_space(self) -> gym.Space: method render (line 81) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method _increment_num_steps_taken (line 100) | def _increment_num_steps_taken(self) -> None: method step (line 105) | def step(self, action: Any) -> RLStepResult: method _step (line 143) | def _step(self, action: Any) -> RLStepResult: method reached_max_steps (line 167) | def reached_max_steps(self) -> bool: method reached_terminal_state (line 172) | def reached_terminal_state(self) -> bool: method is_done (line 177) | def is_done(self) -> bool: method num_steps_taken (line 182) | def num_steps_taken(self) -> int: method action_names (line 187) | def action_names(self) -> Tuple[str, ...]: method close (line 211) | def close(self) -> None: method metrics (line 216) | def metrics(self) -> Dict[str, Any]: method query_expert (line 233) | def query_expert(self, **kwargs) -> Tuple[Any, bool]: method cumulative_reward (line 252) | def cumulative_reward(self) -> float: class TaskSampler (line 269) | class TaskSampler(abc.ABC): method length (line 274) | def length(self) -> Union[int, float]: method last_sampled_task (line 286) | def last_sampled_task(self) -> Optional[Task]: method next_task (line 296) | def next_task(self, force_advance_scene: bool = False) -> Optional[Task]: method close (line 313) | def close(self) -> None: method all_observation_spaces_equal (line 322) | def all_observation_spaces_equal(self) -> bool: method reset (line 340) | def reset(self) -> None: method set_seed (line 345) | def set_seed(self, seed: int) -> None: FILE: allenact/embodiedai/aux_losses/losses.py function _bernoulli_subsample_mask_like (line 28) | def _bernoulli_subsample_mask_like(masks, p=0.1): class MultiAuxTaskNegEntropyLoss (line 32) | class MultiAuxTaskNegEntropyLoss(AbstractActorCriticLoss): method __init__ (line 40) | def __init__(self, task_names: Sequence[str], *args, **kwargs): method loss (line 45) | def loss( # type: ignore class AuxiliaryLoss (line 72) | class AuxiliaryLoss(AbstractActorCriticLoss): method __init__ (line 79) | def __init__(self, auxiliary_uuid: str, *args, **kwargs): method loss (line 84) | def loss( # type: ignore method get_aux_loss (line 102) | def get_aux_loss( function _propagate_final_beliefs_to_all_steps (line 116) | def _propagate_final_beliefs_to_all_steps( class InverseDynamicsLoss (line 146) | class InverseDynamicsLoss(AuxiliaryLoss): method __init__ (line 154) | def __init__( method get_aux_loss (line 165) | def get_aux_loss( class TemporalDistanceLoss (line 251) | class TemporalDistanceLoss(AuxiliaryLoss): method __init__ (line 258) | def __init__(self, num_pairs: int = 8, epsiode_len_min: int = 5, *args... method get_aux_loss (line 263) | def get_aux_loss( class CPCALoss (line 373) | class CPCALoss(AuxiliaryLoss): method __init__ (line 381) | def __init__( method get_aux_loss (line 389) | def get_aux_loss( class CPCASoftMaxLoss (line 567) | class CPCASoftMaxLoss(AuxiliaryLoss): method __init__ (line 572) | def __init__( method get_aux_loss (line 588) | def get_aux_loss( class CPCA1SoftMaxLoss (line 707) | class CPCA1SoftMaxLoss(CPCASoftMaxLoss): method __init__ (line 710) | def __init__(self, subsample_rate: float = 1, *args, **kwargs): class CPCA2SoftMaxLoss (line 716) | class CPCA2SoftMaxLoss(CPCASoftMaxLoss): method __init__ (line 719) | def __init__(self, subsample_rate: float = 1, *args, **kwargs): class CPCA4SoftMaxLoss (line 725) | class CPCA4SoftMaxLoss(CPCASoftMaxLoss): method __init__ (line 728) | def __init__(self, subsample_rate: float = 1, *args, **kwargs): class CPCA8SoftMaxLoss (line 734) | class CPCA8SoftMaxLoss(CPCASoftMaxLoss): method __init__ (line 737) | def __init__(self, subsample_rate: float = 1, *args, **kwargs): class CPCA16SoftMaxLoss (line 743) | class CPCA16SoftMaxLoss(CPCASoftMaxLoss): method __init__ (line 746) | def __init__(self, subsample_rate: float = 1, *args, **kwargs): class CPCA1Loss (line 755) | class CPCA1Loss(CPCALoss): method __init__ (line 758) | def __init__(self, subsample_rate: float = 0.2, *args, **kwargs): class CPCA2Loss (line 764) | class CPCA2Loss(CPCALoss): method __init__ (line 767) | def __init__(self, subsample_rate: float = 0.2, *args, **kwargs): class CPCA4Loss (line 773) | class CPCA4Loss(CPCALoss): method __init__ (line 776) | def __init__(self, subsample_rate: float = 0.2, *args, **kwargs): class CPCA8Loss (line 782) | class CPCA8Loss(CPCALoss): method __init__ (line 785) | def __init__(self, subsample_rate: float = 0.2, *args, **kwargs): class CPCA16Loss (line 791) | class CPCA16Loss(CPCALoss): method __init__ (line 794) | def __init__(self, subsample_rate: float = 0.2, *args, **kwargs): FILE: allenact/embodiedai/mapping/mapping_losses.py class BinnedPointCloudMapLoss (line 12) | class BinnedPointCloudMapLoss(AbstractActorCriticLoss): method __init__ (line 16) | def __init__( method loss (line 38) | def loss( # type: ignore class SemanticMapFocalLoss (line 81) | class SemanticMapFocalLoss(AbstractActorCriticLoss): method __init__ (line 92) | def __init__( method loss (line 114) | def loss( # type: ignore FILE: allenact/embodiedai/mapping/mapping_models/active_neural_slam.py function _inv_sigmoid (line 39) | def _inv_sigmoid(x: torch.Tensor): class ActiveNeuralSLAM (line 43) | class ActiveNeuralSLAM(nn.Module): method __init__ (line 65) | def __init__( method device (line 234) | def device(self): method train (line 240) | def train(self, mode: bool = True): method map_size (line 247) | def map_size(self): method vision_range (line 251) | def vision_range(self): method image_to_egocentric_map_logits (line 254) | def image_to_egocentric_map_logits( method allocentric_map_to_egocentric_view (line 283) | def allocentric_map_to_egocentric_view( method estimate_egocentric_dx_dz_dr (line 334) | def estimate_egocentric_dx_dz_dr( method update_allocentric_xzrs_with_egocentric_movement (line 363) | def update_allocentric_xzrs_with_egocentric_movement( method forward (line 389) | def forward( function _move_egocentric_map_view_into_allocentric_position (line 536) | def _move_egocentric_map_view_into_allocentric_position( FILE: allenact/embodiedai/mapping/mapping_utils/map_builders.py class BinnedPointCloudMapBuilder (line 39) | class BinnedPointCloudMapBuilder(object): method __init__ (line 71) | def __init__( method update (line 102) | def update( method reset (line 298) | def reset(self, min_xyz: np.ndarray): class ObjectHull2d (line 312) | class ObjectHull2d: method __init__ (line 313) | def __init__( class SemanticMapBuilder (line 337) | class SemanticMapBuilder(object): method __init__ (line 359) | def __init__( method randomly_color_semantic_map (line 399) | def randomly_color_semantic_map( method _xzs_to_colrows (line 421) | def _xzs_to_colrows(self, xzs: np.ndarray): method build_ground_truth_map (line 436) | def build_ground_truth_map(self, object_hulls: Sequence[ObjectHull2d]): method update (line 457) | def update( method reset (line 608) | def reset(self, min_xyz: np.ndarray, object_hulls: Sequence[ObjectHull... FILE: allenact/embodiedai/mapping/mapping_utils/point_cloud_utils.py function camera_space_xyz_to_world_xyz (line 34) | def camera_space_xyz_to_world_xyz( function depth_frame_to_camera_space_xyz (line 108) | def depth_frame_to_camera_space_xyz( function depth_frame_to_world_space_xyz (line 164) | def depth_frame_to_world_space_xyz( function project_point_cloud_to_map (line 221) | def project_point_cloud_to_map( function _cpu_only_camera_space_xyz_to_world_xyz (line 336) | def _cpu_only_camera_space_xyz_to_world_xyz( function _cpu_only_depth_frame_to_camera_space_xyz (line 386) | def _cpu_only_depth_frame_to_camera_space_xyz( function _cpu_only_depth_frame_to_world_space_xyz (line 425) | def _cpu_only_depth_frame_to_world_space_xyz( function _cpu_only_project_point_cloud_to_map (line 446) | def _cpu_only_project_point_cloud_to_map( FILE: allenact/embodiedai/models/aux_models.py class AuxiliaryModel (line 21) | class AuxiliaryModel(nn.Module): method __init__ (line 25) | def __init__( method initialize_model_given_aux_uuid (line 46) | def initialize_model_given_aux_uuid(self, aux_uuid: str): method init_inverse_dynamics (line 58) | def init_inverse_dynamics(self): method init_temporal_distance (line 63) | def init_temporal_distance(self): method init_cpca (line 66) | def init_cpca(self): method init_cpca_softmax (line 83) | def init_cpca_softmax(self): method forward (line 105) | def forward(self, features: torch.FloatTensor): FILE: allenact/embodiedai/models/basic_models.py class SimpleCNN (line 29) | class SimpleCNN(nn.Module): method __init__ (line 41) | def __init__( method make_cnn_from_params (line 113) | def make_cnn_from_params( method _conv_output_dim (line 156) | def _conv_output_dim( method layer_init (line 198) | def layer_init(cnn) -> None: method is_blind (line 207) | def is_blind(self): method forward (line 212) | def forward(self, observations: Dict[str, torch.Tensor]): # type: ignore class RNNStateEncoder (line 246) | class RNNStateEncoder(nn.Module): method __init__ (line 253) | def __init__( method layer_init (line 290) | def layer_init(self): method num_recurrent_layers (line 299) | def num_recurrent_layers(self) -> int: method _pack_hidden (line 303) | def _pack_hidden( method _unpack_hidden (line 320) | def _unpack_hidden( method _mask_hidden (line 333) | def _mask_hidden( method single_forward (line 375) | def single_forward( method adapt_input (line 414) | def adapt_input( method adapt_result (line 469) | def adapt_result( method seq_forward (line 495) | def seq_forward( # type: ignore method forward (line 567) | def forward( # type: ignore class LinearActorCritic (line 581) | class LinearActorCritic(ActorCriticModel[CategoricalDistr]): method __init__ (line 582) | def __init__( method _recurrent_memory_specification (line 609) | def _recurrent_memory_specification(self): method forward (line 612) | def forward(self, observations, memory, prev_actions, masks): class RNNActorCritic (line 628) | class RNNActorCritic(ActorCriticModel[Distr]): method __init__ (line 629) | def __init__( method recurrent_hidden_state_size (line 681) | def recurrent_hidden_state_size(self) -> int: method num_recurrent_layers (line 685) | def num_recurrent_layers(self) -> int: method _recurrent_memory_specification (line 688) | def _recurrent_memory_specification(self): method forward (line 700) | def forward( # type:ignore FILE: allenact/embodiedai/models/fusion_models.py class Fusion (line 14) | class Fusion(nn.Module): method __init__ (line 20) | def __init__(self, hidden_size, obs_embed_size, num_tasks): method forward (line 26) | def forward( method get_belief_weights (line 52) | def get_belief_weights( class AverageFusion (line 60) | class AverageFusion(Fusion): method get_belief_weights (line 63) | def get_belief_weights( class SoftmaxFusion (line 75) | class SoftmaxFusion(Fusion): method __init__ (line 81) | def __init__(self, hidden_size, obs_embed_size, num_tasks): method get_belief_weights (line 87) | def get_belief_weights( class AttentiveFusion (line 98) | class AttentiveFusion(Fusion): method __init__ (line 104) | def __init__(self, hidden_size, obs_embed_size, num_tasks): method get_belief_weights (line 108) | def get_belief_weights( FILE: allenact/embodiedai/models/resnet.py function conv3x3 (line 19) | def conv3x3(in_planes, out_planes, stride=1, groups=1): function conv1x1 (line 32) | def conv1x1(in_planes, out_planes, stride=1): class BasicBlock (line 37) | class BasicBlock(nn.Module): method __init__ (line 41) | def __init__( method forward (line 61) | def forward(self, x): function _build_bottleneck_branch (line 72) | def _build_bottleneck_branch(inplanes, planes, ngroups, stride, expansio... class SE (line 85) | class SE(nn.Module): method __init__ (line 86) | def __init__(self, planes, r=16): method forward (line 96) | def forward(self, x): function _build_se_branch (line 105) | def _build_se_branch(planes, r=16): class Bottleneck (line 109) | class Bottleneck(nn.Module): method __init__ (line 113) | def __init__( method _impl (line 134) | def _impl(self, x): method forward (line 144) | def forward(self, x): class SEBottleneck (line 148) | class SEBottleneck(Bottleneck): method __init__ (line 149) | def __init__( method _impl (line 162) | def _impl(self, x): class SEResNeXtBottleneck (line 174) | class SEResNeXtBottleneck(SEBottleneck): class ResNeXtBottleneck (line 179) | class ResNeXtBottleneck(Bottleneck): class GroupNormResNet (line 184) | class GroupNormResNet(nn.Module): method __init__ (line 185) | def __init__(self, in_channels, base_planes, ngroups, block, layers, c... method _make_layer (line 220) | def _make_layer(self, block, ngroups, planes, blocks, stride=1): method forward (line 244) | def forward(self, x): function gnresnet18 (line 256) | def gnresnet18(in_channels, base_planes, ngroups): function gnresnet50 (line 262) | def gnresnet50(in_channels, base_planes, ngroups): function gnresneXt50 (line 268) | def gnresneXt50(in_channels, base_planes, ngroups): function se_gnresnet50 (line 281) | def se_gnresnet50(in_channels, base_planes, ngroups): function se_gnresneXt50 (line 289) | def se_gnresneXt50(in_channels, base_planes, ngroups): function se_gnresneXt101 (line 302) | def se_gnresneXt101(in_channels, base_planes, ngroups): class GroupNormResNetEncoder (line 315) | class GroupNormResNetEncoder(nn.Module): method __init__ (line 316) | def __init__( method is_blind (line 392) | def is_blind(self): method layer_init (line 395) | def layer_init(self): method forward (line 403) | def forward(self, observations): FILE: allenact/embodiedai/models/visual_nav_models.py class VisualNavActorCritic (line 29) | class VisualNavActorCritic(ActorCriticModel[CategoricalDistr]): method __init__ (line 38) | def __init__( method create_state_encoders (line 68) | def create_state_encoders( method load_state_dict (line 122) | def load_state_dict(self, state_dict, **kwargs): method create_actorcritic_head (line 137) | def create_actorcritic_head(self): method create_aux_models (line 141) | def create_aux_models(self, obs_embed_size: int, action_embed_size: int): method num_recurrent_layers (line 157) | def num_recurrent_layers(self): method recurrent_hidden_state_size (line 162) | def recurrent_hidden_state_size(self): method _recurrent_memory_specification (line 166) | def _recurrent_memory_specification(self): method forward_encoder (line 179) | def forward_encoder(self, observations: ObservationType) -> torch.Floa... method fuse_beliefs (line 182) | def fuse_beliefs( method forward (line 195) | def forward( # type:ignore FILE: allenact/embodiedai/preprocessors/resnet.py class ResNetEmbedder (line 13) | class ResNetEmbedder(nn.Module): method __init__ (line 14) | def __init__(self, resnet, pool=True): method forward (line 20) | def forward(self, x): class ResNetPreprocessor (line 40) | class ResNetPreprocessor(Preprocessor): method __init__ (line 43) | def __init__( method resnet (line 93) | def resnet(self) -> ResNetEmbedder: method to (line 100) | def to(self, device: torch.device) -> "ResNetPreprocessor": method process (line 105) | def process(self, obs: Dict[str, Any], *args: Any, **kwargs: Any) -> Any: FILE: allenact/embodiedai/sensors/vision_sensors.py class VisionSensor (line 19) | class VisionSensor(Sensor[EnvType, SubTaskType]): method __init__ (line 20) | def __init__( method _make_observation_space (line 95) | def _make_observation_space( method _get_observation_space (line 137) | def _get_observation_space(self): method height (line 141) | def height(self) -> Optional[int]: method width (line 151) | def width(self) -> Optional[int]: method frame_from_env (line 161) | def frame_from_env(self, env: EnvType, task: Optional[SubTaskType]) ->... method process_img (line 164) | def process_img(self, img: np.ndarray): method get_observation (line 200) | def get_observation( class RGBSensor (line 206) | class RGBSensor(VisionSensor[EnvType, SubTaskType], ABC): method __init__ (line 207) | def __init__( class DepthSensor (line 255) | class DepthSensor(VisionSensor[EnvType, SubTaskType], ABC): method __init__ (line 256) | def __init__( method get_observation (line 306) | def get_observation( # type: ignore FILE: allenact/embodiedai/storage/vdr_storage.py function _index_recursive (line 23) | def _index_recursive(d: Dict, key: Union[str, Tuple[str, ...]]): class InverseDynamicsVDRLoss (line 31) | class InverseDynamicsVDRLoss(GenericAbstractLoss): method __init__ (line 32) | def __init__( method loss (line 44) | def loss( class DiscreteVisualDynamicsReplayStorage (line 68) | class DiscreteVisualDynamicsReplayStorage(ExperienceStorage, MiniBatchSt... method __init__ (line 69) | def __init__( method total_experiences (line 98) | def total_experiences(self): method set_partition (line 101) | def set_partition(self, index: int, num_parts: int): method initialize (line 107) | def initialize(self, *, observations: ObservationType, **kwargs): method batched_experience_generator (line 111) | def batched_experience_generator(self, num_mini_batch: int): method add (line 135) | def add( method before_updates (line 211) | def before_updates(self, **kwargs): method after_updates (line 214) | def after_updates(self, **kwargs): method to (line 217) | def to(self, device: torch.device): FILE: allenact/main.py function get_argument_parser (line 29) | def get_argument_parser(): function get_args (line 335) | def get_args(): function _config_source (line 351) | def _config_source(config_type: Type) -> Dict[str, str]: function find_sub_modules (line 367) | def find_sub_modules(path: str, module_list: Optional[List] = None): function load_config (line 383) | def load_config(args) -> Tuple[ExperimentConfig, Dict[str, str]]: function main (line 463) | def main(): FILE: allenact/setup.py function parse_req_file (line 7) | def parse_req_file(fname, initial=None): function get_version (line 44) | def get_version(fname): function _do_setup (line 59) | def _do_setup(): FILE: allenact/utils/cache_utils.py function pos_to_str_for_cache (line 7) | def pos_to_str_for_cache(pos: Dict[str, float]) -> str: function str_to_pos_for_cache (line 11) | def str_to_pos_for_cache(s: str) -> Dict[str, float]: function get_distance (line 16) | def get_distance( function get_distance_to_object (line 58) | def get_distance_to_object( function _get_shortest_path_distance_from_cache (line 96) | def _get_shortest_path_distance_from_cache( function _get_shortest_path_distance_to_object_from_cache (line 107) | def _get_shortest_path_distance_to_object_from_cache( function find_nearest_point_in_cache (line 116) | def find_nearest_point_in_cache( class DynamicDistanceCache (line 134) | class DynamicDistanceCache(object): method __init__ (line 135) | def __init__(self, rounding: Optional[int] = None): method find_distance (line 142) | def find_distance( method invalidate (line 173) | def invalidate(self): method _pos_to_str (line 176) | def _pos_to_str(self, pos: Dict[str, Any]) -> str: FILE: allenact/utils/cacheless_frcnn.py class CachelessAnchorGenerator (line 13) | class CachelessAnchorGenerator(AnchorGenerator): method forward (line 14) | def forward(self, image_list: Any, feature_maps: Any): function fasterrcnn_resnet50_fpn (line 34) | def fasterrcnn_resnet50_fpn( FILE: allenact/utils/experiment_utils.py function evenly_distribute_count_into_bins (line 54) | def evenly_distribute_count_into_bins(count: int, nbins: int) -> List[int]: function recursive_update (line 72) | def recursive_update( class Builder (line 98) | class Builder(tuple, Generic[ToBuildType]): method __new__ (line 120) | def __new__( method __repr__ (line 145) | def __repr__(self) -> str: method __call__ (line 152) | def __call__(self, **kwargs) -> ToBuildType: class ScalarMeanTracker (line 172) | class ScalarMeanTracker(object): method __init__ (line 175) | def __init__(self) -> None: method add_scalars (line 179) | def add_scalars( method pop_and_reset (line 200) | def pop_and_reset(self) -> Dict[str, float]: method reset (line 216) | def reset(self): method sums (line 220) | def sums(self): method counts (line 223) | def counts(self) -> Dict[str, int]: method means (line 226) | def means(self) -> Dict[str, float]: method empty (line 232) | def empty(self): class LoggingPackage (line 241) | class LoggingPackage: method __init__ (line 244) | def __init__( method num_non_empty_metrics_dicts_added (line 271) | def num_non_empty_metrics_dicts_added(self) -> int: method _metrics_dict_is_empty (line 275) | def _metrics_dict_is_empty( method add_metrics_dict (line 290) | def add_metrics_dict( method add_info_dict (line 303) | def add_info_dict( class LinearDecay (line 318) | class LinearDecay(object): method __init__ (line 331) | def __init__(self, steps: int, startp: float = 1.0, endp: float = 0.0)... method __call__ (line 340) | def __call__(self, epoch: int) -> float: class MultiLinearDecay (line 355) | class MultiLinearDecay(object): method __init__ (line 367) | def __init__(self, stages: Sequence[LinearDecay]) -> None: method __call__ (line 380) | def __call__(self, epoch: int) -> float: function set_deterministic_cudnn (line 405) | def set_deterministic_cudnn() -> None: function set_seed (line 415) | def set_seed(seed: Optional[int] = None) -> None: class EarlyStoppingCriterion (line 432) | class EarlyStoppingCriterion(abc.ABC): method __call__ (line 437) | def __call__( class NeverEarlyStoppingCriterion (line 457) | class NeverEarlyStoppingCriterion(EarlyStoppingCriterion): method __call__ (line 460) | def __call__( class OffPolicyPipelineComponent (line 469) | class OffPolicyPipelineComponent(NamedTuple): class TrainingSettings (line 496) | class TrainingSettings: method __init__ (line 532) | def __init__( method keys (line 550) | def keys(self) -> Tuple[str, ...]: method has_key (line 553) | def has_key(self, key: str) -> bool: method set_defaults (line 556) | def set_defaults(self, defaults: "TrainingSettings"): method __getattr__ (line 560) | def __getattr__(self, item: str): class StageComponent (line 571) | class StageComponent: method _validate_training_settings (line 594) | def _validate_training_settings(self, attribute, value: TrainingSettin... class PipelineStage (line 612) | class PipelineStage: method __init__ (line 639) | def __init__( method reset (line 704) | def reset(self): method change_engine_attributes (line 716) | def change_engine_attributes(self, engine: Any): method stage_components (line 727) | def stage_components(self) -> Tuple[StageComponent]: method add_stage_component (line 730) | def add_stage_component(self, stage_component: StageComponent): method __setattr__ (line 765) | def __setattr__(self, key: str, value: Any): method is_complete (line 778) | def is_complete(self): class TrainingPipeline (line 785) | class TrainingPipeline: method __init__ (line 815) | def __init__( method _initialize_rollout_storage_uuid (line 901) | def _initialize_rollout_storage_uuid( method _ensure_pipeline_stages_all_have_at_least_one_stage_component (line 918) | def _ensure_pipeline_stages_all_have_at_least_one_stage_component(self): method _get_uuids_of_rollout_storages (line 967) | def _get_uuids_of_rollout_storages( method _initialize_named_storages (line 982) | def _initialize_named_storages( method _refresh_current_stage (line 999) | def _refresh_current_stage( method total_steps (line 1019) | def total_steps(self) -> int: method storage_uuid_to_total_experiences (line 1023) | def storage_uuid_to_total_experiences(self) -> Dict[str, int]: method current_stage (line 1045) | def current_stage(self) -> Optional[PipelineStage]: method current_stage_index (line 1049) | def current_stage_index(self) -> Optional[int]: method before_rollout (line 1054) | def before_rollout(self, train_metrics: Optional[ScalarMeanTracker] = ... method restart_pipeline (line 1075) | def restart_pipeline(self): method state_dict (line 1088) | def state_dict(self): method load_state_dict (line 1102) | def load_state_dict(self, state_dict: Dict[str, Any]): method rollout_storage (line 1127) | def rollout_storage(self) -> Optional[RolloutStorage]: method get_stage_storage (line 1138) | def get_stage_storage( method current_stage_storage (line 1165) | def current_stage_storage(self) -> "OrderedDict[str, ExperienceStorage]": method get_loss (line 1168) | def get_loss(self, uuid: str): method current_stage_losses (line 1177) | def current_stage_losses( function download_checkpoint_from_wandb (line 1196) | def download_checkpoint_from_wandb( FILE: allenact/utils/inference.py class InferenceAgent (line 21) | class InferenceAgent: method __attrs_post_init__ (line 32) | def __attrs_post_init__(self): method from_experiment_config (line 44) | def from_experiment_config( method reset (line 88) | def reset(self): method act (line 94) | def act(self, observations: ObservationType): FILE: allenact/utils/misc_utils.py function multiprocessing_safe_download_file_from_url (line 39) | def multiprocessing_safe_download_file_from_url(url: str, save_path: str): function experimental_api (line 51) | def experimental_api(to_decorate): function deprecated (line 74) | def deprecated(to_decorate): class NumpyJSONEncoder (line 95) | class NumpyJSONEncoder(json.JSONEncoder): method default (line 102) | def default(self, obj): function tensor_print_options (line 118) | def tensor_print_options(**print_opts): function md5_hash_str_as_int (line 130) | def md5_hash_str_as_int(to_hash: str): function get_git_diff_of_project (line 137) | def get_git_diff_of_project() -> Tuple[str, str]: class HashableDict (line 145) | class HashableDict(dict): method __init__ (line 155) | def __init__(self, *args, **kwargs): method __key (line 160) | def __key(self): method __hash__ (line 163) | def __hash__(self): method __eq__ (line 167) | def __eq__(self, other): method __setitem__ (line 170) | def __setitem__(self, *args, **kwargs): method __delitem__ (line 175) | def __delitem__(self, *args, **kwargs): function partition_sequence (line 183) | def partition_sequence(seq: Sequence, parts: int) -> List: function unzip (line 195) | def unzip(seq: Sequence[Tuple], n: Optional[int]): function uninterleave (line 222) | def uninterleave(seq: Sequence, parts: int) -> List: function cached_comb (line 235) | def cached_comb(n: int, m: int): function expected_max_of_subset_statistic (line 239) | def expected_max_of_subset_statistic(vals: List[float], m: int): function bootstrap_max_of_subset_statistic (line 266) | def bootstrap_max_of_subset_statistic( function rand_float (line 284) | def rand_float(low: float, high: float, shape): function all_unique (line 292) | def all_unique(seq: Sequence[Hashable]): function all_equal (line 301) | def all_equal(s: Sequence): function prepare_locals_for_super (line 307) | def prepare_locals_for_super( function partition_limits (line 324) | def partition_limits(num_items: int, num_parts: int): function str2bool (line 332) | def str2bool(v: str): class ForkedPdb (line 342) | class ForkedPdb(pdb.Pdb): method interaction (line 345) | def interaction(self, *args, **kwargs): FILE: allenact/utils/model_utils.py function md5_hash_of_state_dict (line 13) | def md5_hash_of_state_dict(state_dict: Dict[str, Any]): class Flatten (line 34) | class Flatten(nn.Module): method forward (line 38) | def forward(self, x): function init_linear_layer (line 49) | def init_linear_layer( function grad_norm (line 72) | def grad_norm(parameters, norm_type=2): function make_cnn (line 88) | def make_cnn( function compute_cnn_output (line 144) | def compute_cnn_output( function simple_conv_and_linear_weights_init (line 210) | def simple_conv_and_linear_weights_init(m): function simple_linear_weights_init (line 230) | def simple_linear_weights_init(m): class FeatureEmbedding (line 241) | class FeatureEmbedding(nn.Module): method __init__ (line 245) | def __init__(self, input_size, output_size): method forward (line 261) | def forward(self, inputs): FILE: allenact/utils/multi_agent_viz_utils.py class MultiTrajectoryViz (line 10) | class MultiTrajectoryViz(TrajectoryViz): method __init__ (line 11) | def __init__( method make_fig (line 29) | def make_fig(self, episode, episode_id): FILE: allenact/utils/spaces_utils.py function flatdim (line 16) | def flatdim(space): function flatten (line 40) | def flatten(space, torch_x): function unflatten (line 69) | def unflatten(space, torch_x): function torch_point (line 100) | def torch_point(space, np_x): function numpy_point (line 120) | def numpy_point( function flatten_space (line 150) | def flatten_space(space: gym.Space): function policy_space (line 177) | def policy_space( function action_list (line 222) | def action_list( FILE: allenact/utils/system.py class ColoredFormatter (line 21) | class ColoredFormatter(logging.Formatter): method __init__ (line 42) | def __init__(self, fmt: str, datefmt: Optional[str] = None, use_color=... method format (line 46) | def format(self, record: logging.LogRecord) -> str: function get_logger (line 64) | def get_logger() -> logging.Logger: function _human_log_level_to_int (line 81) | def _human_log_level_to_int(human_log_level): function init_logging (line 103) | def init_logging(human_log_level: str = "info") -> None: function update_log_level (line 114) | def update_log_level(logger, human_log_level: str): function find_free_port (line 118) | def find_free_port(address: str = "127.0.0.1") -> int: function _new_logger (line 132) | def _new_logger(log_level: Optional[int] = None): function _set_log_formatter (line 144) | def _set_log_formatter(): class _StreamToLogger (line 191) | class _StreamToLogger: method __init__ (line 192) | def __init__(self): method write (line 195) | def write(self, buf): method flush (line 204) | def flush(self): function _excepthook (line 210) | def _excepthook(*args): class _AllenActMessageFilter (line 215) | class _AllenActMessageFilter: method __init__ (line 216) | def __init__(self, working_directory: str): method filter (line 220) | def filter(self, record): class ImportChecker (line 229) | class ImportChecker: method __init__ (line 230) | def __init__(self, msg=None): method __enter__ (line 233) | def __enter__(self): method __exit__ (line 236) | def __exit__(self, exc_type, value, traceback): FILE: allenact/utils/tensor_utils.py function to_device_recursively (line 25) | def to_device_recursively( function detach_recursively (line 84) | def detach_recursively(input: Any, inplace=True): function batch_observations (line 130) | def batch_observations( function to_tensor (line 191) | def to_tensor(v) -> torch.Tensor: function tile_images (line 212) | def tile_images(images: List[np.ndarray]) -> np.ndarray: class SummaryWriter (line 242) | class SummaryWriter(TBXSummaryWriter): method _video (line 244) | def _video(tag, vid): method add_vid (line 249) | def add_vid(self, tag, vid, global_step=None, walltime=None): method add_image (line 254) | def add_image( function image (line 262) | def image(tag, tensor, rescale=1, dataformats="CHW"): function convert_to_HWC (line 300) | def convert_to_HWC(tensor, input_format): # tensor: numpy array function make_grid (line 336) | def make_grid(I, ncols=8): function tensor_to_video (line 359) | def tensor_to_video(tensor, fps=4): function tensor_to_clip (line 369) | def tensor_to_clip(tensor, fps=4): function clips_to_video (line 383) | def clips_to_video(clips, h, w, c): function process_video (line 411) | def process_video(render, max_clip_len=500, max_video_len=-1, fps=4): class ScaleBothSides (line 463) | class ScaleBothSides(object): method __init__ (line 472) | def __init__(self, width: int, height: int, interpolation=Image.BILINE... method __call__ (line 477) | def __call__(self, img: PIL.Image) -> PIL.Image: FILE: allenact/utils/viz_utils.py class AbstractViz (line 56) | class AbstractViz: method __init__ (line 57) | def __init__( method _source_to_str (line 85) | def _source_to_str(source, is_vector_task): method _access (line 93) | def _access(dictionary, path): method _auto_viz_order (line 99) | def _auto_viz_order(self, task_outputs): method _setup (line 124) | def _setup( method log (line 144) | def log( class TrajectoryViz (line 154) | class TrajectoryViz(AbstractViz): method __init__ (line 155) | def __init__( method log (line 191) | def log( method make_fig (line 224) | def make_fig(self, episode, episode_id): class AgentViewViz (line 321) | class AgentViewViz(AbstractViz): method __init__ (line 322) | def __init__( method log (line 356) | def log( method _overlay_label (line 399) | def _overlay_label( method make_vid (line 430) | def make_vid(self, images): class AbstractTensorViz (line 462) | class AbstractTensorViz(AbstractViz): method __init__ (line 463) | def __init__( method log (line 483) | def log( method make_fig (line 528) | def make_fig( class TensorViz1D (line 534) | class TensorViz1D(AbstractTensorViz): method __init__ (line 535) | def __init__( method make_fig (line 544) | def make_fig(self, episode_src, episode_id): class TensorViz2D (line 560) | class TensorViz2D(AbstractTensorViz): method __init__ (line 561) | def __init__( method make_fig (line 572) | def make_fig(self, episode_src, episode_id): class ActorViz (line 595) | class ActorViz(AbstractViz): method __init__ (line 596) | def __init__( method log (line 612) | def log( method make_fig (line 664) | def make_fig(self, episode_src, episode_id): class VizSuite (line 693) | class VizSuite(AbstractViz): method __init__ (line 694) | def __init__( method _setup_sources (line 740) | def _setup_sources(self): method _episodes_set (line 796) | def _episodes_set(self, episode_list=None) -> Optional[Set[str]]: method empty (line 806) | def empty(self): method _update (line 809) | def _update(self, collected_data): method _append (line 814) | def _append(self, vector_task_data): method _collect_actor_critic (line 821) | def _collect_actor_critic(self, actor_critic): method _collect_rollout (line 868) | def _collect_rollout(self, rollout, alive): method _collect_vector_task (line 942) | def _collect_vector_task(self, vector_task): method collect (line 985) | def collect(self, vector_task=None, alive=None, rollout=None, actor_cr... method read_and_reset (line 1020) | def read_and_reset(self) -> Dict[str, List[Dict[str, Any]]]: method log (line 1027) | def log( class TensorboardSummarizer (line 1045) | class TensorboardSummarizer: method __init__ (line 1048) | def __init__( method _read_tensorflow_experiment_events (line 1093) | def _read_tensorflow_experiment_events( method _eval_vs_train_time_steps (line 1129) | def _eval_vs_train_time_steps(self, eval_data, train_data): method _train_vs_time_steps (line 1160) | def _train_vs_time_steps(self, train_data): method make_tensorboard_summary (line 1180) | def make_tensorboard_summary(self): FILE: allenact_plugins/babyai_plugin/babyai_models.py class BabyAIACModelWrapped (line 22) | class BabyAIACModelWrapped(babyai.model.ACModel): method __init__ (line 23) | def __init__( method forward_once (line 74) | def forward_once(self, obs, memory, instr_embedding=None): method forward_loop (line 138) | def forward_loop( method forward (line 255) | def forward( method adapt_inputs (line 471) | def adapt_inputs( # type: ignore method adapt_result (line 530) | def adapt_result(ac_output, hidden_states, num_steps, num_samplers, nu... class BabyAIRecurrentACModel (line 571) | class BabyAIRecurrentACModel(ActorCriticModel[CategoricalDistr]): method __init__ (line 572) | def __init__( method recurrent_hidden_state_size (line 613) | def recurrent_hidden_state_size(self) -> int: method num_recurrent_layers (line 617) | def num_recurrent_layers(self): method _recurrent_memory_specification (line 620) | def _recurrent_memory_specification(self): method forward (line 632) | def forward( # type:ignore FILE: allenact_plugins/babyai_plugin/babyai_tasks.py class BabyAITask (line 18) | class BabyAITask(Task[MiniGridEnv]): method __init__ (line 19) | def __init__( method action_space (line 45) | def action_space(self) -> gym.spaces.Discrete: method render (line 48) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method _step (line 51) | def _step(self, action: int) -> RLStepResult: method get_observations (line 66) | def get_observations( method reached_terminal_state (line 73) | def reached_terminal_state(self) -> bool: method class_action_names (line 77) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 85) | def close(self) -> None: method _expert_timeout_hander (line 88) | def _expert_timeout_hander(self, signum, frame): method query_expert (line 91) | def query_expert(self, **kwargs) -> Tuple[Any, bool]: method metrics (line 116) | def metrics(self) -> Dict[str, Any]: class BabyAITaskSampler (line 124) | class BabyAITaskSampler(TaskSampler): method __init__ (line 125) | def __init__( method length (line 180) | def length(self) -> Union[int, float]: method total_unique (line 188) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 192) | def last_sampled_task(self) -> Optional[Task]: method next_task (line 195) | def next_task(self, force_advance_scene: bool = False) -> Optional[Bab... method close (line 219) | def close(self) -> None: method all_observation_spaces_equal (line 223) | def all_observation_spaces_equal(self) -> bool: method reset (line 226) | def reset(self) -> None: method set_seed (line 230) | def set_seed(self, seed: int) -> None: FILE: allenact_plugins/babyai_plugin/scripts/download_babyai_expert_demos.py function get_args (line 26) | def get_args(): FILE: allenact_plugins/babyai_plugin/scripts/truncate_expert_demos.py function make_small_demos (line 11) | def make_small_demos(dir: str): FILE: allenact_plugins/clip_plugin/clip_preprocessors.py class ClipResNetEmbedder (line 14) | class ClipResNetEmbedder(nn.Module): method __init__ (line 15) | def __init__(self, resnet: CLIP, pool=True, pooling_type="avg"): method forward (line 34) | def forward(self, x): class ClipResNetPreprocessor (line 39) | class ClipResNetPreprocessor(Preprocessor): method __init__ (line 46) | def __init__( method resnet (line 100) | def resnet(self) -> ClipResNetEmbedder: method to (line 111) | def to(self, device: torch.device) -> "ClipResNetPreprocessor": method process (line 116) | def process(self, obs: Dict[str, Any], *args: Any, **kwargs: Any) -> Any: class ClipViTEmbedder (line 135) | class ClipViTEmbedder(nn.Module): method __init__ (line 136) | def __init__(self, model: CLIP, class_emb_only: bool = False): method forward (line 146) | def forward(self, x): class ClipViTPreprocessor (line 175) | class ClipViTPreprocessor(Preprocessor): method __init__ (line 182) | def __init__( method vit (line 231) | def vit(self) -> ClipViTEmbedder: method to (line 243) | def to(self, device: torch.device) -> "ClipViTPreprocessor": method process (line 248) | def process(self, obs: Dict[str, Any], *args: Any, **kwargs: Any) -> Any: FILE: allenact_plugins/gym_plugin/gym_distributions.py class GaussianDistr (line 6) | class GaussianDistr(torch.distributions.Normal, Distr): method mode (line 9) | def mode(self) -> torch.FloatTensor: FILE: allenact_plugins/gym_plugin/gym_environment.py class GymEnvironment (line 7) | class GymEnvironment(gym.Wrapper): method __init__ (line 10) | def __init__(self, gym_env_name: str): method reset (line 15) | def reset(self) -> np.ndarray: method initial_observation (line 20) | def initial_observation(self) -> np.ndarray: FILE: allenact_plugins/gym_plugin/gym_models.py class MemorylessActorCritic (line 15) | class MemorylessActorCritic(ActorCriticModel[GaussianDistr]): method __init__ (line 19) | def __init__( method make_mlp_hidden (line 58) | def make_mlp_hidden(nl, *dims): method _recurrent_memory_specification (line 67) | def _recurrent_memory_specification(self): method forward (line 70) | def forward( # type:ignore FILE: allenact_plugins/gym_plugin/gym_sensors.py class GymBox2DSensor (line 11) | class GymBox2DSensor(Sensor[gym.Env, Task[gym.Env]]): method __init__ (line 14) | def __init__( method _get_observation_space (line 26) | def _get_observation_space(self) -> gym.Space: method get_observation (line 39) | def get_observation( class GymMuJoCoSensor (line 53) | class GymMuJoCoSensor(Sensor[gym.Env, Task[gym.Env]]): method __init__ (line 56) | def __init__(self, gym_env_name: str, uuid: str, **kwargs: Any): method _get_observation_space (line 63) | def _get_observation_space(self) -> gym.Space: method get_observation (line 97) | def get_observation( FILE: allenact_plugins/gym_plugin/gym_tasks.py class GymTask (line 17) | class GymTask(Task[gym.Env]): method __init__ (line 23) | def __init__( method action_space (line 38) | def action_space(self) -> gym.spaces.Space: method render (line 41) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method get_observations (line 46) | def get_observations( method reached_terminal_state (line 53) | def reached_terminal_state(self) -> bool: method close (line 56) | def close(self) -> None: method metrics (line 59) | def metrics(self) -> Dict[str, Any]: class GymContinuousTask (line 73) | class GymContinuousTask(GymTask): method class_action_names (line 78) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method _step (line 81) | def _step(self, action: Sequence[float]) -> RLStepResult: function default_task_selector (line 94) | def default_task_selector(env_name: str) -> type: function sensor_selector (line 117) | def sensor_selector(env_name: str) -> Sensor: class GymTaskSampler (line 142) | class GymTaskSampler(TaskSampler): method __init__ (line 145) | def __init__( method length (line 231) | def length(self) -> Union[int, float]: method total_unique (line 239) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 243) | def last_sampled_task(self) -> Optional[Task]: method next_task (line 246) | def next_task(self, force_advance_scene: bool = False) -> Optional[Gym... method close (line 298) | def close(self) -> None: method all_observation_spaces_equal (line 302) | def all_observation_spaces_equal(self) -> bool: method reset (line 305) | def reset(self) -> None: method set_seed (line 309) | def set_seed(self, seed: int) -> None: FILE: allenact_plugins/habitat_plugin/habitat_environment.py class HabitatEnvironment (line 17) | class HabitatEnvironment: method __init__ (line 18) | def __init__(self, config: Config, dataset: Dataset, verbose: bool = F... method scene_name (line 32) | def scene_name(self) -> str: method current_frame (line 36) | def current_frame(self) -> np.ndarray: method step (line 40) | def step(self, action_dict: Dict[str, Union[str, int]]) -> Observations: method get_location (line 45) | def get_location(self) -> Optional[np.ndarray]: method get_rotation (line 48) | def get_rotation(self) -> Optional[List[float]]: method get_shortest_path (line 51) | def get_shortest_path( method get_current_episode (line 58) | def get_current_episode(self) -> HabitatNavigationEpisode: method start (line 62) | def start(self): method stop (line 65) | def stop(self): method reset (line 68) | def reset(self): method last_action_success (line 72) | def last_action_success(self) -> bool: method num_episodes (line 77) | def num_episodes(self) -> int: FILE: allenact_plugins/habitat_plugin/habitat_sensors.py class RGBSensorHabitat (line 17) | class RGBSensorHabitat(RGBSensor[HabitatEnvironment, Task[HabitatEnviron... method __init__ (line 19) | def __init__( method frame_from_env (line 40) | def frame_from_env( class DepthSensorHabitat (line 46) | class DepthSensorHabitat(DepthSensor[HabitatEnvironment, Task[HabitatEnv... method __init__ (line 48) | def __init__( method frame_from_env (line 72) | def frame_from_env( class TargetCoordinatesSensorHabitat (line 78) | class TargetCoordinatesSensorHabitat(Sensor[HabitatEnvironment, "PointNa... method __init__ (line 79) | def __init__( method _get_observation_space (line 88) | def _get_observation_space(self): method get_observation (line 94) | def get_observation( class TargetObjectSensorHabitat (line 106) | class TargetObjectSensorHabitat(Sensor[HabitatEnvironment, "ObjectNavTas... method __init__ (line 107) | def __init__(self, num_objects: int, uuid: str = "target_object_id", *... method _get_observation_space (line 112) | def _get_observation_space(num_objects: int): method get_observation (line 115) | def get_observation( class AgentCoordinatesSensorHabitat (line 127) | class AgentCoordinatesSensorHabitat(Sensor[HabitatEnvironment, "PointNav... method __init__ (line 128) | def __init__(self, uuid: str = "agent_position_and_rotation", **kwargs... method _get_observation_space (line 134) | def _get_observation_space(): method get_observation (line 138) | def get_observation( FILE: allenact_plugins/habitat_plugin/habitat_task_samplers.py class PointNavTaskSampler (line 14) | class PointNavTaskSampler(TaskSampler): method __init__ (line 15) | def __init__( method _create_environment (line 43) | def _create_environment(self) -> HabitatEnvironment: method length (line 61) | def length(self) -> Union[int, float]: method total_unique (line 68) | def total_unique(self) -> Union[int, float, None]: method last_sampled_task (line 72) | def last_sampled_task(self) -> Optional[PointNavTask]: method close (line 75) | def close(self) -> None: method all_observation_spaces_equal (line 80) | def all_observation_spaces_equal(self) -> bool: method next_task (line 87) | def next_task(self, force_advance_scene=False) -> Optional[PointNavTask]: method reset (line 122) | def reset(self): method set_seed (line 125) | def set_seed(self, seed: int): class ObjectNavTaskSampler (line 131) | class ObjectNavTaskSampler(TaskSampler): method __init__ (line 132) | def __init__( method _create_environment (line 162) | def _create_environment(self) -> HabitatEnvironment: method length (line 180) | def length(self) -> Union[int, float]: method total_unique (line 187) | def total_unique(self) -> Union[int, float, None]: method last_sampled_task (line 191) | def last_sampled_task(self) -> Optional[ObjectNavTask]: method close (line 194) | def close(self) -> None: method all_observation_spaces_equal (line 199) | def all_observation_spaces_equal(self) -> bool: method next_task (line 206) | def next_task(self, force_advance_scene=False) -> Optional[ObjectNavTa... method reset (line 246) | def reset(self): method set_seed (line 249) | def set_seed(self, seed: int): FILE: allenact_plugins/habitat_plugin/habitat_tasks.py class HabitatTask (line 28) | class HabitatTask(Task[HabitatEnvironment], ABC): method __init__ (line 29) | def __init__( method get_agent_position_and_rotation (line 55) | def get_agent_position_and_rotation(self): method last_action (line 59) | def last_action(self): method last_action (line 63) | def last_action(self, value: str): method last_action_success (line 67) | def last_action_success(self): method last_action_success (line 71) | def last_action_success(self, value: Optional[bool]): method render (line 74) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: class PointNavTask (line 83) | class PointNavTask(Task[HabitatEnvironment]): method __init__ (line 86) | def __init__( method current_geodesic_dist_to_target (line 118) | def current_geodesic_dist_to_target(self) -> Optional[float]: method action_space (line 130) | def action_space(self): method reached_terminal_state (line 133) | def reached_terminal_state(self) -> bool: method class_action_names (line 137) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 140) | def close(self) -> None: method _step (line 143) | def _step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method render (line 166) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method _is_goal_in_range (line 170) | def _is_goal_in_range(self) -> bool: method judge (line 175) | def judge(self) -> float: method metrics (line 202) | def metrics(self) -> Dict[str, Any]: method query_expert (line 218) | def query_expert(self, **kwargs) -> Tuple[int, bool]: class ObjectNavTask (line 234) | class ObjectNavTask(HabitatTask): method __init__ (line 237) | def __init__( method action_space (line 278) | def action_space(self): method reached_terminal_state (line 281) | def reached_terminal_state(self) -> bool: method class_action_names (line 285) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method action_names (line 288) | def action_names(self, **kwargs) -> Tuple[str, ...]: method close (line 291) | def close(self) -> None: method current_geodesic_dist_to_target (line 294) | def current_geodesic_dist_to_target(self) -> Optional[float]: method _step (line 305) | def _step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method render (line 359) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method _is_goal_in_range (line 363) | def _is_goal_in_range(self) -> bool: method judge (line 367) | def judge(self) -> float: method metrics (line 394) | def metrics(self) -> Dict[str, Any]: method query_expert (line 414) | def query_expert(self, **kwargs) -> Tuple[int, bool]: FILE: allenact_plugins/habitat_plugin/habitat_utils.py function construct_env_configs (line 12) | def construct_env_configs( function construct_env_configs_mp3d (line 77) | def construct_env_configs_mp3d(config: Config) -> List[Config]: function distribute (line 213) | def distribute( function get_habitat_config (line 227) | def get_habitat_config(path: str): FILE: allenact_plugins/habitat_plugin/scripts/agent_demo.py function transform_rgb_bgr (line 20) | def transform_rgb_bgr(image): function agent_demo (line 24) | def agent_demo(): FILE: allenact_plugins/habitat_plugin/scripts/make_map.py function make_map (line 19) | def make_map(env, scene): function generate_maps (line 33) | def generate_maps(): FILE: allenact_plugins/ithor_plugin/ithor_constants.py function make_scene_name (line 22) | def make_scene_name(type_ind, scene_num): FILE: allenact_plugins/ithor_plugin/ithor_environment.py class IThorEnvironment (line 20) | class IThorEnvironment(object): method __init__ (line 32) | def __init__( method scene_name (line 104) | def scene_name(self) -> str: method current_frame (line 109) | def current_frame(self) -> np.ndarray: method last_event (line 114) | def last_event(self) -> ai2thor.server.Event: method started (line 119) | def started(self) -> bool: method last_action (line 124) | def last_action(self) -> str: method last_action (line 129) | def last_action(self, value: str) -> None: method last_action_success (line 137) | def last_action_success(self) -> bool: method last_action_success (line 142) | def last_action_success(self, value: bool) -> None: method last_action_return (line 150) | def last_action_return(self) -> Any: method last_action_return (line 159) | def last_action_return(self, value: Any) -> None: method start (line 166) | def start( method stop (line 212) | def stop(self) -> None: method reset (line 221) | def reset( method teleport_agent_to (line 273) | def teleport_agent_to( method random_reachable_state (line 378) | def random_reachable_state(self, seed: int = None) -> Dict: method randomize_agent_location (line 390) | def randomize_agent_location( method object_pixels_in_frame (line 418) | def object_pixels_in_frame( method object_pixels_on_grid (line 454) | def object_pixels_on_grid( method object_in_hand (line 494) | def object_in_hand(self): method initially_reachable_points (line 507) | def initially_reachable_points(self) -> List[Dict[str, float]]: method initially_reachable_points_set (line 514) | def initially_reachable_points_set(self) -> Set[Tuple[float, float]]: method currently_reachable_points (line 527) | def currently_reachable_points(self) -> List[Dict[str, float]]: method get_agent_location (line 533) | def get_agent_location(self) -> Dict[str, Union[float, bool]]: method _agent_location_to_tuple (line 547) | def _agent_location_to_tuple(p: Dict[str, float]) -> Tuple[float, float]: method _snap_agent_to_initially_reachable (line 550) | def _snap_agent_to_initially_reachable(self, verbose=True): method step (line 648) | def step( method position_dist (line 733) | def position_dist( method rotation_dist (line 754) | def rotation_dist(a: Dict[str, float], b: Dict[str, float]): method angle_between_rotations (line 764) | def angle_between_rotations(a: Dict[str, float], b: Dict[str, float]): method closest_object_with_properties (line 773) | def closest_object_with_properties( method closest_visible_object_of_type (line 794) | def closest_visible_object_of_type( method closest_object_of_type (line 802) | def closest_object_of_type(self, object_type: str) -> Optional[Dict[st... method closest_reachable_point_to_position (line 807) | def closest_reachable_point_to_position( method _angle_from_to (line 826) | def _angle_from_to(a_from: float, a_to: float) -> float: method agent_xz_to_scene_xz (line 840) | def agent_xz_to_scene_xz(self, agent_xz: Dict[str, float]) -> Dict[str... method scene_xz_to_agent_xz (line 865) | def scene_xz_to_agent_xz(self, scene_xz: Dict[str, float]) -> Dict[str... method all_objects (line 888) | def all_objects(self) -> List[Dict[str, Any]]: method all_objects_with_properties (line 892) | def all_objects_with_properties( method visible_objects (line 907) | def visible_objects(self) -> List[Dict[str, Any]]: method get_object_by_id (line 911) | def get_object_by_id(self, object_id: str) -> Optional[Dict[str, Any]]: method reachable_points_with_rotations_and_horizons (line 924) | def reachable_points_with_rotations_and_horizons(self): method location_for_key (line 941) | def location_for_key(key, y_value=0.0): method get_key (line 947) | def get_key(input_dict: Dict[str, Any]) -> Tuple[float, float, int, int]: method update_graph_with_failed_action (line 966) | def update_graph_with_failed_action(self, failed_action: str): method _add_from_to_edge (line 985) | def _add_from_to_edge( method possible_neighbor_offsets (line 1031) | def possible_neighbor_offsets(self) -> Tuple[Tuple[float, float, int, ... method _add_node_to_graph (line 1044) | def _add_node_to_graph(self, graph: nx.DiGraph, s: Tuple[float, float,... method graph (line 1058) | def graph(self): method graph (line 1069) | def graph(self, g): method _check_contains_key (line 1072) | def _check_contains_key(self, key: Tuple[float, float, int, int], add_... method shortest_state_path (line 1080) | def shortest_state_path(self, source_state_key, goal_state_key): method action_transitioning_between_keys (line 1090) | def action_transitioning_between_keys(self, s, t): method shortest_path_next_state (line 1098) | def shortest_path_next_state(self, source_state_key, goal_state_key): method shortest_path_next_action (line 1106) | def shortest_path_next_action(self, source_state_key, goal_state_key): method shortest_path_length (line 1113) | def shortest_path_length(self, source_state_key, goal_state_key): FILE: allenact_plugins/ithor_plugin/ithor_sensors.py class RGBSensorThor (line 36) | class RGBSensorThor(RGBSensor[THOR_ENV_TYPE, THOR_TASK_TYPE]): method frame_from_env (line 43) | def frame_from_env( class GoalObjectTypeThorSensor (line 54) | class GoalObjectTypeThorSensor(Sensor): method __init__ (line 55) | def __init__( method _get_observation_space (line 91) | def _get_observation_space(self): method get_observation (line 97) | def get_observation( class TakeEndActionThorNavSensor (line 107) | class TakeEndActionThorNavSensor( method __init__ (line 113) | def __init__(self, nactions: int, uuid: str, **kwargs: Any) -> None: method _get_observation_space (line 120) | def _get_observation_space(self) -> gym.spaces.Discrete: method get_observation (line 129) | def get_observation( # type:ignore class RelativePositionChangeTHORSensor (line 150) | class RelativePositionChangeTHORSensor( method __init__ (line 153) | def __init__(self, uuid: str = "rel_position_change", **kwargs: Any): method get_relative_position_change (line 175) | def get_relative_position_change(from_xzr: np.ndarray, to_xzr: np.ndar... method get_observation (line 200) | def get_observation( class ReachableBoundsTHORSensor (line 228) | class ReachableBoundsTHORSensor(Sensor[RoboThorEnvironment, Task[RoboTho... method __init__ (line 229) | def __init__(self, margin: float, uuid: str = "scene_bounds", **kwargs... method get_bounds (line 252) | def get_bounds( method get_observation (line 267) | def get_observation( class SceneBoundsTHORSensor (line 288) | class SceneBoundsTHORSensor(Sensor[RoboThorEnvironment, Task[RoboThorEnv... method __init__ (line 289) | def __init__(self, uuid: str = "scene_bounds", **kwargs: Any): method get_observation (line 308) | def get_observation( class BinnedPointCloudMapTHORSensor (line 329) | class BinnedPointCloudMapTHORSensor( method __init__ (line 334) | def __init__( method device (line 399) | def device(self): method device (line 403) | def device(self, val: torch.device): method get_observation (line 406) | def get_observation( class SemanticMapTHORSensor (line 453) | class SemanticMapTHORSensor(Sensor[RoboThorEnvironment, Task[RoboThorEnv... method __init__ (line 456) | def __init__( method device (line 522) | def device(self): method device (line 526) | def device(self, val: torch.device): method get_observation (line 529) | def get_observation( FILE: allenact_plugins/ithor_plugin/ithor_task_samplers.py class ObjectNavTaskSampler (line 15) | class ObjectNavTaskSampler(TaskSampler): method __init__ (line 16) | def __init__( method _create_environment (line 58) | def _create_environment(self) -> IThorEnvironment: method length (line 68) | def length(self) -> Union[int, float]: method total_unique (line 78) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 82) | def last_sampled_task(self) -> Optional[ObjectNaviThorGridTask]: method close (line 85) | def close(self) -> None: method all_observation_spaces_equal (line 90) | def all_observation_spaces_equal(self) -> bool: method sample_scene (line 100) | def sample_scene(self, force_advance_scene: bool): method next_task (line 141) | def next_task( method reset (line 190) | def reset(self): method set_seed (line 197) | def set_seed(self, seed: int): FILE: allenact_plugins/ithor_plugin/ithor_tasks.py class ObjectNaviThorGridTask (line 24) | class ObjectNaviThorGridTask(Task[IThorEnvironment]): method __init__ (line 62) | def __init__( method action_space (line 87) | def action_space(self): method reached_terminal_state (line 90) | def reached_terminal_state(self) -> bool: method class_action_names (line 94) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 97) | def close(self) -> None: method _step (line 100) | def _step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method render (line 129) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method is_goal_object_visible (line 133) | def is_goal_object_visible(self) -> bool: method judge (line 140) | def judge(self) -> float: method metrics (line 152) | def metrics(self) -> Dict[str, Any]: method query_expert (line 161) | def query_expert(self, **kwargs) -> Tuple[int, bool]: FILE: allenact_plugins/ithor_plugin/ithor_util.py function include_object_data (line 16) | def include_object_data(controller: ai2thor.controller.Controller): function vertical_to_horizontal_fov (line 29) | def vertical_to_horizontal_fov( function horizontal_to_vertical_fov (line 42) | def horizontal_to_vertical_fov( function round_to_factor (line 52) | def round_to_factor(num: float, base: int) -> int: function get_open_x_displays (line 65) | def get_open_x_displays(throw_error_if_empty: bool = False) -> Sequence[... FILE: allenact_plugins/ithor_plugin/ithor_viz.py class ThorPositionTo2DFrameTranslator (line 23) | class ThorPositionTo2DFrameTranslator(object): method __init__ (line 24) | def __init__( method __call__ (line 34) | def __call__(self, position: Sequence[float]): class ThorViz (line 50) | class ThorViz(TrajectoryViz): method __init__ (line 51) | def __init__( method init_top_down_render (line 99) | def init_top_down_render(self): method iterate_scenes (line 109) | def iterate_scenes( method cached_map_data_path (line 117) | def cached_map_data_path(self, roomname: str) -> str: method get_translator (line 120) | def get_translator(self) -> Dict[str, Any]: method cached_image_path (line 147) | def cached_image_path(self, roomname: str) -> str: method make_top_down_views (line 152) | def make_top_down_views(self) -> Dict[str, np.ndarray]: method crop_viz_image (line 163) | def crop_viz_image(self, viz_image: np.ndarray) -> np.ndarray: method make_controller (line 172) | def make_controller(self): method get_agent_map_data (line 185) | def get_agent_map_data(self): method position_to_tuple (line 197) | def position_to_tuple(position: Dict[str, float]) -> Tuple[float, floa... method add_lines_to_map (line 201) | def add_lines_to_map( method add_line_to_map (line 230) | def add_line_to_map( method add_agent_view_triangle (line 258) | def add_agent_view_triangle( method visualize_agent_path (line 292) | def visualize_agent_path( method dump_top_down_view (line 346) | def dump_top_down_view(self, room_name: str, image_path: str): method make_fig (line 358) | def make_fig(self, episode: Any, episode_id: str) -> Figure: class ThorMultiViz (line 386) | class ThorMultiViz(ThorViz): method __init__ (line 387) | def __init__( method make_fig (line 404) | def make_fig(self, episode: Any, episode_id: str) -> Figure: FILE: allenact_plugins/lighthouse_plugin/lighthouse_environment.py function _get_world_corners (line 19) | def _get_world_corners(world_dim: int, world_radius: int): function _base_world_tensor (line 39) | def _base_world_tensor(world_dim: int, world_radius: int): class LightHouseEnvironment (line 56) | class LightHouseEnvironment(object): method __init__ (line 63) | def __init__(self, world_dim: int, world_radius: int, **kwargs): method set_seed (line 91) | def set_seed(self, seed: int): method random_reset (line 101) | def random_reset(self, goal_position: Optional[bool] = None): method step (line 129) | def step(self, action: int) -> bool: method render (line 151) | def render(self, mode="array", **kwargs): method close (line 256) | def close(self): method optimal_ave_ep_length (line 264) | def optimal_ave_ep_length(world_dim: int, world_radius: int, view_radi... FILE: allenact_plugins/lighthouse_plugin/lighthouse_models.py class LinearAdvisorActorCritic (line 17) | class LinearAdvisorActorCritic(ActorCriticModel[CategoricalDistr]): method __init__ (line 18) | def __init__( method _recurrent_memory_specification (line 52) | def _recurrent_memory_specification(self): method forward (line 55) | def forward( # type:ignore FILE: allenact_plugins/lighthouse_plugin/lighthouse_sensors.py function get_corner_observation (line 16) | def get_corner_observation( class CornerSensor (line 76) | class CornerSensor(Sensor[LightHouseEnvironment, Any]): method __init__ (line 77) | def __init__( method _get_observation_space (line 92) | def _get_observation_space(self): method get_observation (line 100) | def get_observation( class FactorialDesignCornerSensor (line 119) | class FactorialDesignCornerSensor(Sensor[LightHouseEnvironment, Any]): method __init__ (line 122) | def __init__( method _get_observation_space (line 177) | def _get_observation_space(self): method view_tuple_to_design_array (line 191) | def view_tuple_to_design_array(self, view_tuple: Tuple): method output_dim (line 197) | def output_dim(cls, world_dim: int): method _create_full_design_matrix_and_tuple_to_ind_dict (line 203) | def _create_full_design_matrix_and_tuple_to_ind_dict( method _get_variables_and_levels (line 241) | def _get_variables_and_levels(world_dim: int): method _create_formula (line 252) | def _create_formula( method get_observation (line 272) | def get_observation( FILE: allenact_plugins/lighthouse_plugin/lighthouse_tasks.py class LightHouseTask (line 24) | class LightHouseTask(Task[LightHouseEnvironment], abc.ABC): method __init__ (line 36) | def __init__( method last_action (line 55) | def last_action(self) -> int: method last_action (line 59) | def last_action(self, value: int): method step (line 62) | def step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method render (line 69) | def render(self, mode: str = "array", *args, **kwargs) -> np.ndarray: class FindGoalLightHouseTask (line 94) | class FindGoalLightHouseTask(LightHouseTask): method __init__ (line 97) | def __init__( method action_space (line 110) | def action_space(self) -> gym.spaces.Discrete: method _step (line 113) | def _step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method reached_terminal_state (line 133) | def reached_terminal_state(self) -> bool: method class_action_names (line 137) | def class_action_names(cls, world_dim: int = 2, **kwargs) -> Tuple[str... method action_names (line 150) | def action_names(self) -> Tuple[str, ...]: method close (line 153) | def close(self) -> None: method query_expert (line 156) | def query_expert( class FindGoalLightHouseTaskSampler (line 311) | class FindGoalLightHouseTaskSampler(TaskSampler): method __init__ (line 312) | def __init__( method world_dim (line 370) | def world_dim(self): method world_radius (line 374) | def world_radius(self): method length (line 378) | def length(self) -> Union[int, float]: method total_unique (line 386) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 391) | def last_sampled_task(self) -> Optional[Task]: method next_task (line 394) | def next_task(self, force_advance_scene: bool = False) -> Optional[Task]: method close (line 416) | def close(self) -> None: method all_observation_spaces_equal (line 420) | def all_observation_spaces_equal(self) -> bool: method reset (line 423) | def reset(self) -> None: method set_seed (line 427) | def set_seed(self, seed: int) -> None: FILE: allenact_plugins/lighthouse_plugin/lighthouse_util.py class StopIfNearOptimal (line 6) | class StopIfNearOptimal(EarlyStoppingCriterion): method __init__ (line 7) | def __init__(self, optimal: float, deviation: float, min_memory_size: ... method __call__ (line 15) | def __call__( FILE: allenact_plugins/manipulathor_plugin/arm_calculation_utils.py function state_dict_to_tensor (line 13) | def state_dict_to_tensor(state: Dict): function diff_position (line 30) | def diff_position(state_goal, state_curr, absolute: bool = True): function coord_system_transform (line 40) | def coord_system_transform(position: Dict, coord_system: str): function position_rotation_to_matrix (line 84) | def position_rotation_to_matrix(position, rotation): function inverse_rot_trans_matrix (line 93) | def inverse_rot_trans_matrix(mat): function matrix_to_position_rotation (line 98) | def matrix_to_position_rotation(matrix): function find_closest_inverse (line 108) | def find_closest_inverse(deg, use_cache): function calc_inverse (line 122) | def calc_inverse(deg): function world_coords_to_agent_coords (line 133) | def world_coords_to_agent_coords(world_obj, agent_state, use_cache=True): FILE: allenact_plugins/manipulathor_plugin/armpointnav_constants.py function get_agent_start_positions (line 22) | def get_agent_start_positions(): FILE: allenact_plugins/manipulathor_plugin/manipulathor_environment.py function position_distance (line 46) | def position_distance(s1, s2, filter_nan: bool = False): function rotation_distance (line 59) | def rotation_distance(s1: Dict[str, Dict[str, float]], s2: Dict[str, Dic... class ManipulaTHOREnvironment (line 71) | class ManipulaTHOREnvironment(IThorEnvironment): method __init__ (line 83) | def __init__( method create_controller (line 137) | def create_controller(self): method start (line 142) | def start( method reset (line 168) | def reset( method randomize_agent_location (line 208) | def randomize_agent_location( method is_object_at_low_level_hand (line 213) | def is_object_at_low_level_hand(self, object_id): method object_in_hand (line 219) | def object_in_hand(self): method correct_nan_inf (line 232) | def correct_nan_inf(cls, flawed_dict, extra_tag=""): method get_object_by_id (line 239) | def get_object_by_id(self, object_id: str) -> Optional[Dict[str, Any]]: method get_current_arm_state (line 246) | def get_current_arm_state(self): method get_absolute_hand_state (line 263) | def get_absolute_hand_state(self): method get_pickupable_objects (line 272) | def get_pickupable_objects(self): method get_current_object_locations (line 279) | def get_current_object_locations(self): method close_enough (line 290) | def close_enough(self, current_obj_pose, init_obj_pose, threshold): method get_objects_moved (line 299) | def get_objects_moved( method get_objects_move_distance (line 328) | def get_objects_move_distance( method step (line 374) | def step( FILE: allenact_plugins/manipulathor_plugin/manipulathor_sensors.py class DepthSensorThor (line 23) | class DepthSensorThor( method frame_from_env (line 35) | def frame_from_env( class NoVisionSensorThor (line 41) | class NoVisionSensorThor( method frame_from_env (line 53) | def frame_from_env( class AgentRelativeCurrentObjectStateThorSensor (line 59) | class AgentRelativeCurrentObjectStateThorSensor(Sensor): method __init__ (line 60) | def __init__(self, uuid: str = "relative_current_obj_state", **kwargs:... method get_observation (line 66) | def get_observation( class RelativeObjectToGoalSensor (line 83) | class RelativeObjectToGoalSensor(Sensor): method __init__ (line 84) | def __init__( method get_observation (line 106) | def get_observation( class InitialObjectToGoalSensor (line 127) | class InitialObjectToGoalSensor(Sensor): method __init__ (line 128) | def __init__(self, uuid: str = "initial_obj_to_goal", **kwargs: Any): method get_observation (line 135) | def get_observation( class DistanceObjectToGoalSensor (line 151) | class DistanceObjectToGoalSensor(Sensor): method __init__ (line 152) | def __init__(self, uuid: str = "distance_obj_to_goal", **kwargs: Any): method get_observation (line 159) | def get_observation( class RelativeAgentArmToObjectSensor (line 177) | class RelativeAgentArmToObjectSensor(Sensor): method __init__ (line 178) | def __init__( method get_observation (line 200) | def get_observation( class InitialAgentArmToObjectSensor (line 222) | class InitialAgentArmToObjectSensor(Sensor): method __init__ (line 223) | def __init__(self, uuid: str = "initial_agent_arm_to_obj", **kwargs: A... method get_observation (line 229) | def get_observation( class DistanceAgentArmToObjectSensor (line 248) | class DistanceAgentArmToObjectSensor(Sensor): method __init__ (line 249) | def __init__(self, uuid: str = "distance_agent_arm_to_obj", **kwargs: ... method get_observation (line 255) | def get_observation( class PickedUpObjSensor (line 275) | class PickedUpObjSensor(Sensor): method __init__ (line 276) | def __init__(self, uuid: str = "pickedup_object", **kwargs: Any): method get_observation (line 282) | def get_observation( FILE: allenact_plugins/manipulathor_plugin/manipulathor_task_samplers.py class AbstractMidLevelArmTaskSampler (line 33) | class AbstractMidLevelArmTaskSampler(TaskSampler): method __init__ (line 37) | def __init__( method _create_environment (line 89) | def _create_environment(self, **kwargs) -> ManipulaTHOREnvironment: method last_sampled_task (line 99) | def last_sampled_task(self) -> Optional[Task]: method close (line 102) | def close(self) -> None: method all_observation_spaces_equal (line 107) | def all_observation_spaces_equal(self) -> bool: method reset (line 117) | def reset(self): method set_seed (line 126) | def set_seed(self, seed: int): class SimpleArmPointNavGeneralSampler (line 132) | class SimpleArmPointNavGeneralSampler(AbstractMidLevelArmTaskSampler): method __init__ (line 136) | def __init__(self, **kwargs) -> None: method next_task (line 208) | def next_task( method total_unique (line 291) | def total_unique(self) -> Optional[Union[int, float]]: method length (line 298) | def length(self) -> Union[int, float]: method get_source_target_indices (line 311) | def get_source_target_indices(self): method calc_possible_trajectories (line 328) | def calc_possible_trajectories(self, all_possible_points): class ArmPointNavTaskSampler (line 340) | class ArmPointNavTaskSampler(SimpleArmPointNavGeneralSampler): method __init__ (line 343) | def __init__(self, **kwargs) -> None: method next_task (line 356) | def next_task( method get_source_target_indices (line 449) | def get_source_target_indices(self): class RotateArmPointNavTaskSampler (line 510) | class RotateArmPointNavTaskSampler(ArmPointNavTaskSampler): class CamRotateArmPointNavTaskSampler (line 514) | class CamRotateArmPointNavTaskSampler(ArmPointNavTaskSampler): class EasyArmPointNavTaskSampler (line 518) | class EasyArmPointNavTaskSampler(ArmPointNavTaskSampler): function get_all_tuples_from_list (line 522) | def get_all_tuples_from_list(list): FILE: allenact_plugins/manipulathor_plugin/manipulathor_tasks.py class AbstractPickUpDropOffTask (line 46) | class AbstractPickUpDropOffTask(Task[ManipulaTHOREnvironment]): method __init__ (line 69) | def __init__( method action_space (line 108) | def action_space(self): method reached_terminal_state (line 111) | def reached_terminal_state(self) -> bool: method class_action_names (line 115) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 118) | def close(self) -> None: method obj_state_aproximity (line 121) | def obj_state_aproximity(self, s1, s2): method start_visualize (line 132) | def start_visualize(self): method visualize (line 140) | def visualize(self, action_str): method finish_visualizer (line 145) | def finish_visualizer(self): method finish_visualizer_metrics (line 150) | def finish_visualizer_metrics(self, metric_results): method render (line 155) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method calc_action_stat_metrics (line 159) | def calc_action_stat_metrics(self) -> Dict[str, Any]: method metrics (line 186) | def metrics(self) -> Dict[str, Any]: method _step (line 238) | def _step(self, action: int) -> RLStepResult: method arm_distance_from_obj (line 241) | def arm_distance_from_obj(self): method obj_distance_from_goal (line 247) | def obj_distance_from_goal(self): method get_original_object_distance (line 253) | def get_original_object_distance(self): method judge (line 261) | def judge(self) -> float: class ArmPointNavTask (line 266) | class ArmPointNavTask(AbstractPickUpDropOffTask): method __init__ (line 283) | def __init__( method metrics (line 307) | def metrics(self) -> Dict[str, Any]: method visualize (line 321) | def visualize(self, **kwargs): method _step (line 326) | def _step(self, action: int) -> RLStepResult: method judge (line 369) | def judge(self) -> float: class RotateArmPointNavTask (line 454) | class RotateArmPointNavTask(ArmPointNavTask): class CamRotateArmPointNavTask (line 478) | class CamRotateArmPointNavTask(ArmPointNavTask): class EasyArmPointNavTask (line 504) | class EasyArmPointNavTask(ArmPointNavTask): method _step (line 521) | def _step(self, action: int) -> RLStepResult: FILE: allenact_plugins/manipulathor_plugin/manipulathor_utils.py function make_all_objects_unbreakable (line 12) | def make_all_objects_unbreakable(controller): function reset_environment_and_additional_commands (line 23) | def reset_environment_and_additional_commands(controller, scene_name): function transport_wrapper (line 31) | def transport_wrapper(controller, target_object, target_location): function initialize_arm (line 51) | def initialize_arm(controller): FILE: allenact_plugins/manipulathor_plugin/manipulathor_viz.py class LoggerVisualizer (line 20) | class LoggerVisualizer: method __init__ (line 21) | def __init__(self, exp_name="", log_dir=""): method log (line 37) | def log(self, environment, action_str): method is_empty (line 40) | def is_empty(self): method finish_episode_metrics (line 43) | def finish_episode_metrics(self, episode_info, task_info, metric_resul... method finish_episode (line 46) | def finish_episode(self, environment, episode_info, task_info): class TestMetricLogger (line 50) | class TestMetricLogger(LoggerVisualizer): method __init__ (line 51) | def __init__(self, exp_name="", log_dir="", **kwargs): method average_dict (line 58) | def average_dict(self): method finish_episode_metrics (line 64) | def finish_episode_metrics(self, episode_info, task_info, metric_resul... method log (line 99) | def log(self, environment, action_str="", disturbance_str=""): class BringObjImageVisualizer (line 105) | class BringObjImageVisualizer(LoggerVisualizer): method finish_episode (line 106) | def finish_episode(self, environment, episode_info, task_info): method log (line 159) | def log(self, environment, action_str): method log_start_goal (line 164) | def log_start_goal(self, env, task_info, tag, img_adr): class ImageVisualizer (line 220) | class ImageVisualizer(LoggerVisualizer): method __init__ (line 221) | def __init__( method finish_episode (line 235) | def finish_episode(self, environment, episode_info, task_info): method log (line 305) | def log(self, environment, action_str="", disturbance_str=""): method log_start_goal (line 320) | def log_start_goal(self, env, task_info, tag, img_adr): function save_image_list_to_gif (line 375) | def save_image_list_to_gif(image_list, gif_name, gif_dir): function put_annotation_on_image (line 393) | def put_annotation_on_image(images, annotations): FILE: allenact_plugins/minigrid_plugin/configs/minigrid_nomemory.py class MiniGridNoMemoryExperimentConfig (line 12) | class MiniGridNoMemoryExperimentConfig(MiniGridTutorialExperimentConfig): method tag (line 14) | def tag(cls) -> str: method create_model (line 18) | def create_model(cls, **kwargs) -> nn.Module: FILE: allenact_plugins/minigrid_plugin/minigrid_environments.py class FastCrossing (line 10) | class FastCrossing(CrossingEnv): method __init__ (line 14) | def __init__(self, size=9, num_crossings=1, obstacle_type=Lava, seed=N... method same_seed_reset (line 25) | def same_seed_reset(self): method reset (line 45) | def reset(self, partial_reset: bool = False): class AskForHelpSimpleCrossing (line 51) | class AskForHelpSimpleCrossing(CrossingEnv): method __init__ (line 54) | def __init__( method was_successful (line 84) | def was_successful(self) -> bool: method gen_obs (line 87) | def gen_obs(self): method metrics (line 93) | def metrics(self): method step (line 104) | def step(self, action: int): method same_seed_reset (line 136) | def same_seed_reset(self): method reset (line 161) | def reset(self, partial_reset: bool = False): class LavaCrossingS25N10 (line 171) | class LavaCrossingS25N10(CrossingEnv): method __init__ (line 172) | def __init__(self): class LavaCrossingS15N7 (line 176) | class LavaCrossingS15N7(CrossingEnv): method __init__ (line 177) | def __init__(self): class LavaCrossingS11N7 (line 181) | class LavaCrossingS11N7(CrossingEnv): method __init__ (line 182) | def __init__(self): FILE: allenact_plugins/minigrid_plugin/minigrid_models.py class MiniGridSimpleConvBase (line 22) | class MiniGridSimpleConvBase(ActorCriticModel[Distr], abc.ABC): method __init__ (line 25) | def __init__( method forward (line 86) | def forward( # type:ignore class MiniGridSimpleConvRNN (line 141) | class MiniGridSimpleConvRNN(MiniGridSimpleConvBase): method __init__ (line 142) | def __init__( method num_recurrent_layers (line 191) | def num_recurrent_layers(self): method recurrent_hidden_state_size (line 195) | def recurrent_hidden_state_size(self): method _recurrent_memory_specification (line 198) | def _recurrent_memory_specification(self): class MiniGridSimpleConv (line 211) | class MiniGridSimpleConv(MiniGridSimpleConvBase): method __init__ (line 212) | def __init__( method num_recurrent_layers (line 250) | def num_recurrent_layers(self): method recurrent_hidden_state_size (line 254) | def recurrent_hidden_state_size(self): method _recurrent_memory_specification (line 258) | def _recurrent_memory_specification(self): FILE: allenact_plugins/minigrid_plugin/minigrid_offpolicy.py class MiniGridOffPolicyExpertCELoss (line 28) | class MiniGridOffPolicyExpertCELoss(GenericAbstractLoss): method __init__ (line 29) | def __init__(self, total_episodes_in_epoch: Optional[int] = None): method loss (line 33) | def loss( # type: ignore function transform_demos (line 91) | def transform_demos(demos): class MiniGridExpertTrajectoryStorage (line 125) | class MiniGridExpertTrajectoryStorage(ExperienceStorage, StreamingStorag... method __init__ (line 126) | def __init__( method data (line 163) | def data(self) -> List[Tuple[str, bytes, List[int], MiniGridEnv.Action... method set_partition (line 203) | def set_partition(self, index: int, num_parts: int): method initialize (line 219) | def initialize(self, *, observations: ObservationType, **kwargs): method add (line 223) | def add( method to (line 235) | def to(self, device: torch.device): method total_experiences (line 239) | def total_experiences(self) -> int: method reset_stream (line 242) | def reset_stream(self): method empty (line 245) | def empty(self) -> bool: method _get_next_ind (line 248) | def _get_next_ind(self): method _fill_rollout_queue (line 254) | def _fill_rollout_queue(self, q: queue.Queue, sampler: int): method get_data_for_rollout_ind (line 266) | def get_data_for_rollout_ind(self, sampler_ind: int) -> Dict[str, np.n... method next_batch (line 319) | def next_batch(self) -> Dict[str, torch.Tensor]: FILE: allenact_plugins/minigrid_plugin/minigrid_sensors.py class EgocentricMiniGridSensor (line 24) | class EgocentricMiniGridSensor(Sensor[MiniGridEnv, Task[MiniGridEnv]]): method __init__ (line 25) | def __init__( method _get_observation_space (line 53) | def _get_observation_space(self) -> gym.Space: method get_observation (line 61) | def get_observation( class MiniGridMissionSensor (line 82) | class MiniGridMissionSensor(Sensor[MiniGridEnv, Task[MiniGridEnv]]): method __init__ (line 83) | def __init__(self, instr_len: int, uuid: str = "minigrid_mission", **k... method _get_observation_space (line 103) | def _get_observation_space(self) -> gym.Space: method get_observation (line 111) | def get_observation( FILE: allenact_plugins/minigrid_plugin/minigrid_tasks.py class MiniGridTask (line 25) | class MiniGridTask(Task[CrossingEnv]): method __init__ (line 50) | def __init__( method action_space (line 72) | def action_space(self) -> gym.spaces.Discrete: method render (line 75) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method _step (line 78) | def _step(self, action: int) -> RLStepResult: method get_observations (line 95) | def get_observations( method reached_terminal_state (line 102) | def reached_terminal_state(self) -> bool: method class_action_names (line 106) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 109) | def close(self) -> None: method metrics (line 112) | def metrics(self) -> Dict[str, Any]: method graph_created (line 126) | def graph_created(self): method graph (line 130) | def graph(self): method graph (line 141) | def graph(self, graph: nx.DiGraph): method possible_neighbor_offsets (line 145) | def possible_neighbor_offsets(cls) -> Tuple[Tuple[int, int, int], ...]: method _add_from_to_edge (line 165) | def _add_from_to_edge( method _add_node_to_graph (line 223) | def _add_node_to_graph( method generate_graph (line 250) | def generate_graph( method query_expert (line 294) | def query_expert(self, **kwargs) -> Tuple[int, bool]: class AskForHelpSimpleCrossingTask (line 350) | class AskForHelpSimpleCrossingTask(MiniGridTask): method __init__ (line 357) | def __init__( method _step (line 371) | def _step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method metrics (line 378) | def metrics(self) -> Dict[str, Any]: class MiniGridTaskSampler (line 387) | class MiniGridTaskSampler(TaskSampler): method __init__ (line 388) | def __init__( method length (line 469) | def length(self) -> Union[int, float]: method total_unique (line 477) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 481) | def last_sampled_task(self) -> Optional[Task]: method next_task (line 484) | def next_task(self, force_advance_scene: bool = False) -> Optional[Min... method close (line 547) | def close(self) -> None: method all_observation_spaces_equal (line 551) | def all_observation_spaces_equal(self) -> bool: method reset (line 554) | def reset(self) -> None: method set_seed (line 558) | def set_seed(self, seed: int) -> None: FILE: allenact_plugins/navigation_plugin/objectnav/models.py class CatObservations (line 23) | class CatObservations(nn.Module): method __init__ (line 24) | def __init__(self, ordered_uuids: Sequence[str], dim: int): method forward (line 31) | def forward(self, observations: ObservationType): class ObjectNavActorCritic (line 39) | class ObjectNavActorCritic(VisualNavActorCritic): method __init__ (line 55) | def __init__( method is_blind (line 170) | def is_blind(self) -> bool: method goal_visual_encoder_output_dims (line 176) | def goal_visual_encoder_output_dims(self): method get_object_type_encoding (line 182) | def get_object_type_encoding( method forward_encoder (line 191) | def forward_encoder(self, observations: ObservationType) -> torch.Tensor: class ResnetTensorNavActorCritic (line 205) | class ResnetTensorNavActorCritic(VisualNavActorCritic): method __init__ (line 206) | def __init__( method is_blind (line 286) | def is_blind(self) -> bool: method forward_encoder (line 291) | def forward_encoder(self, observations: ObservationType) -> torch.Floa... class ResnetTensorGoalEncoder (line 295) | class ResnetTensorGoalEncoder(nn.Module): method __init__ (line 296) | def __init__( method is_blind (line 343) | def is_blind(self): method output_dims (line 347) | def output_dims(self): method get_object_type_encoding (line 357) | def get_object_type_encoding( method compress_resnet (line 366) | def compress_resnet(self, observations): method distribute_target (line 369) | def distribute_target(self, observations): method adapt_input (line 375) | def adapt_input(self, observations): method adapt_output (line 395) | def adapt_output(x, use_agent, nstep, nsampler, nagent): method forward (line 400) | def forward(self, observations): class ResnetDualTensorGoalEncoder (line 422) | class ResnetDualTensorGoalEncoder(nn.Module): method __init__ (line 423) | def __init__( method is_blind (line 492) | def is_blind(self): method output_dims (line 496) | def output_dims(self): method get_object_type_encoding (line 507) | def get_object_type_encoding( method compress_rgb_resnet (line 516) | def compress_rgb_resnet(self, observations): method compress_depth_resnet (line 519) | def compress_depth_resnet(self, observations): method distribute_target (line 522) | def distribute_target(self, observations): method adapt_input (line 528) | def adapt_input(self, observations): method adapt_output (line 548) | def adapt_output(x, use_agent, nstep, nsampler, nagent): method forward (line 553) | def forward(self, observations): FILE: allenact_plugins/navigation_plugin/pointnav/models.py class PointNavActorCritic (line 23) | class PointNavActorCritic(VisualNavActorCritic): method __init__ (line 26) | def __init__( method is_blind (line 115) | def is_blind(self): method goal_visual_encoder_output_dims (line 119) | def goal_visual_encoder_output_dims(self): method get_target_coordinates_encoding (line 125) | def get_target_coordinates_encoding(self, observations): method forward_encoder (line 133) | def forward_encoder(self, observations: ObservationType) -> torch.Floa... FILE: allenact_plugins/robothor_plugin/robothor_distributions.py class TupleCategoricalDistr (line 8) | class TupleCategoricalDistr(Distr): method __init__ (line 9) | def __init__(self, probs=None, logits=None, validate_args=None): method log_prob (line 14) | def log_prob(self, actions: Tuple[torch.LongTensor, ...]) -> torch.Flo... method entropy (line 18) | def entropy(self) -> torch.FloatTensor: method sample (line 22) | def sample(self, sample_shape=torch.Size()) -> Tuple[torch.LongTensor,... method mode (line 27) | def mode(self) -> Tuple[torch.LongTensor, ...]: FILE: allenact_plugins/robothor_plugin/robothor_environment.py class RoboThorEnvironment (line 18) | class RoboThorEnvironment: method __init__ (line 31) | def __init__(self, all_metadata_available: bool = True, **kwargs): method initialize_grid_dimensions (line 79) | def initialize_grid_dimensions( method set_object_filter (line 99) | def set_object_filter(self, object_ids: List[str]): method reset_object_filter (line 102) | def reset_object_filter(self): method path_from_point_to_object_type (line 105) | def path_from_point_to_object_type( method distance_from_point_to_object_type (line 127) | def distance_from_point_to_object_type( method distance_to_object_type (line 145) | def distance_to_object_type(self, object_type: str, agent_id: int = 0)... method path_from_point_to_point (line 189) | def path_from_point_to_point( method distance_from_point_to_point (line 212) | def distance_from_point_to_point( method distance_to_point (line 229) | def distance_to_point(self, target: Dict[str, float], agent_id: int = ... method agent_state (line 271) | def agent_state(self, agent_id: int = 0) -> Dict: method teleport (line 282) | def teleport( method reset (line 311) | def reset( method random_reachable_state (line 332) | def random_reachable_state( method randomize_agent_location (line 357) | def randomize_agent_location( method known_good_locations_list (line 395) | def known_good_locations_list(self): method currently_reachable_points (line 402) | def currently_reachable_points(self) -> List[Dict[str, float]]: method scene_name (line 412) | def scene_name(self) -> str: method current_frame (line 417) | def current_frame(self) -> np.ndarray: method current_depth (line 422) | def current_depth(self) -> np.ndarray: method current_frames (line 427) | def current_frames(self) -> List[np.ndarray]: method current_depths (line 435) | def current_depths(self) -> List[np.ndarray]: method last_event (line 444) | def last_event(self) -> ai2thor.server.Event: method last_action (line 449) | def last_action(self) -> str: method last_action_success (line 454) | def last_action_success(self) -> bool: method last_action_return (line 459) | def last_action_return(self) -> Any: method step (line 467) | def step( method stop (line 479) | def stop(self): method all_objects (line 486) | def all_objects(self) -> List[Dict[str, Any]]: method all_objects_with_properties (line 490) | def all_objects_with_properties( method visible_objects (line 505) | def visible_objects(self) -> List[Dict[str, Any]]: FILE: allenact_plugins/robothor_plugin/robothor_models.py class TupleLinearActorCriticHead (line 21) | class TupleLinearActorCriticHead(LinearActorCriticHead): method forward (line 22) | def forward(self, x): class NavToPartnerActorCriticSimpleConvRNN (line 34) | class NavToPartnerActorCriticSimpleConvRNN(ActorCriticModel[TupleCategor... method __init__ (line 37) | def __init__( method output_size (line 73) | def output_size(self): method is_blind (line 77) | def is_blind(self): method num_recurrent_layers (line 81) | def num_recurrent_layers(self): method recurrent_hidden_state_size (line 85) | def recurrent_hidden_state_size(self): method num_agents (line 89) | def num_agents(self): method _recurrent_memory_specification (line 92) | def _recurrent_memory_specification(self): method forward (line 105) | def forward( # type:ignore FILE: allenact_plugins/robothor_plugin/robothor_preprocessors.py class BatchedFasterRCNN (line 14) | class BatchedFasterRCNN(torch.nn.Module): method __init__ (line 32) | def __init__(self, thres=0.12, maxdets=3, res=7): method detector_tensor (line 41) | def detector_tensor(self, boxes, classes, scores, aspect_ratio=1.0): method forward (line 94) | def forward(self, imbatch): class FasterRCNNPreProcessorRoboThor (line 155) | class FasterRCNNPreProcessorRoboThor(Preprocessor): method __init__ (line 160) | def __init__( method to (line 212) | def to(self, device: torch.device) -> "FasterRCNNPreProcessorRoboThor": method process (line 217) | def process(self, obs: Dict[str, Any], *args: Any, **kwargs: Any) -> Any: FILE: allenact_plugins/robothor_plugin/robothor_sensors.py class RGBSensorRoboThor (line 22) | class RGBSensorRoboThor(RGBSensorThor): method __init__ (line 29) | def __init__(self, *args: Any, **kwargs: Any): class RGBSensorMultiRoboThor (line 36) | class RGBSensorMultiRoboThor(RGBSensor[RoboThorEnvironment, Task[RoboTho... method __init__ (line 43) | def __init__(self, agent_count: int = 2, **kwargs): method frame_from_env (line 49) | def frame_from_env( method get_observation (line 54) | def get_observation( class GPSCompassSensorRoboThor (line 67) | class GPSCompassSensorRoboThor(Sensor[RoboThorEnvironment, PointNavTask]): method __init__ (line 68) | def __init__(self, uuid: str = "target_coordinates_ind", **kwargs: Any): method _get_observation_space (line 73) | def _get_observation_space(self): method _compute_pointgoal (line 82) | def _compute_pointgoal( method quaternion_from_y_angle (line 98) | def quaternion_from_y_angle(angle: float) -> np.quaternion: method quaternion_from_coeff (line 107) | def quaternion_from_coeff(coeffs: np.ndarray) -> np.quaternion: method cartesian_to_polar (line 115) | def cartesian_to_polar(x, y): method quaternion_rotate_vector (line 121) | def quaternion_rotate_vector(quat: np.quaternion, v: np.array) -> np.a... method get_observation (line 133) | def get_observation( class DepthSensorThor (line 154) | class DepthSensorThor( method __init__ (line 160) | def __init__( method frame_from_env (line 184) | def frame_from_env( class DepthSensorRoboThor (line 193) | class DepthSensorRoboThor(DepthSensorThor): method __init__ (line 195) | def __init__(self, *args: Any, **kwargs: Any): FILE: allenact_plugins/robothor_plugin/robothor_task_samplers.py class ObjectNavTaskSampler (line 22) | class ObjectNavTaskSampler(TaskSampler): method __init__ (line 23) | def __init__( method _create_environment (line 98) | def _create_environment(self) -> RoboThorEnvironment: method length (line 103) | def length(self) -> Union[int, float]: method total_unique (line 113) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 117) | def last_sampled_task(self) -> Optional[ObjectNavTask]: method close (line 120) | def close(self) -> None: method all_observation_spaces_equal (line 125) | def all_observation_spaces_equal(self) -> bool: method sample_scene (line 135) | def sample_scene(self, force_advance_scene: bool): method next_task (line 182) | def next_task(self, force_advance_scene: bool = False) -> Optional[Obj... method reset (line 278) | def reset(self): method set_seed (line 286) | def set_seed(self, seed: int): class ObjectNavDatasetTaskSampler (line 292) | class ObjectNavDatasetTaskSampler(TaskSampler): method __init__ (line 293) | def __init__( method _create_environment (line 367) | def _create_environment(self) -> RoboThorEnvironment: method load_dataset (line 372) | def load_dataset(scene: str, base_directory: str) -> List[Dict]: method load_distance_cache_from_file (line 388) | def load_distance_cache_from_file(scene: str, base_directory: str) -> ... method __len__ (line 403) | def __len__(self) -> Union[int, float]: method total_unique (line 413) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 417) | def last_sampled_task(self) -> Optional[ObjectNavTask]: method close (line 420) | def close(self) -> None: method all_observation_spaces_equal (line 425) | def all_observation_spaces_equal(self) -> bool: method length (line 436) | def length(self) -> Union[int, float]: method next_task (line 445) | def next_task(self, force_advance_scene: bool = False) -> Optional[Obj... method reset (line 524) | def reset(self): method set_seed (line 529) | def set_seed(self, seed: int): class PointNavTaskSampler (line 535) | class PointNavTaskSampler(TaskSampler): method __init__ (line 536) | def __init__( method _create_environment (line 583) | def _create_environment(self) -> RoboThorEnvironment: method length (line 588) | def length(self) -> Union[int, float]: method total_unique (line 599) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 607) | def last_sampled_task(self) -> Optional[PointNavTask]: method close (line 610) | def close(self) -> None: method all_observation_spaces_equal (line 615) | def all_observation_spaces_equal(self) -> bool: method sample_scene (line 625) | def sample_scene(self, force_advance_scene: bool): method next_task (line 672) | def next_task(self, force_advance_scene: bool = False) -> Optional[Poi... method reset (line 739) | def reset(self): method set_seed (line 751) | def set_seed(self, seed: int): class PointNavDatasetTaskSampler (line 757) | class PointNavDatasetTaskSampler(TaskSampler): method __init__ (line 758) | def __init__( method _create_environment (line 813) | def _create_environment(self) -> RoboThorEnvironment: method __len__ (line 818) | def __len__(self) -> Union[int, float]: method total_unique (line 828) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 832) | def last_sampled_task(self) -> Optional[PointNavTask]: method close (line 835) | def close(self) -> None: method all_observation_spaces_equal (line 840) | def all_observation_spaces_equal(self) -> bool: method next_task (line 850) | def next_task(self, force_advance_scene: bool = False) -> Optional[Poi... method reset (line 917) | def reset(self): method set_seed (line 922) | def set_seed(self, seed: int): method length (line 928) | def length(self) -> Union[int, float]: class NavToPartnerTaskSampler (line 939) | class NavToPartnerTaskSampler(TaskSampler): method __init__ (line 940) | def __init__( method _create_environment (line 981) | def _create_environment(self) -> RoboThorEnvironment: method length (line 989) | def length(self) -> Union[int, float]: method total_unique (line 1000) | def total_unique(self) -> Optional[Union[int, float]]: method last_sampled_task (line 1004) | def last_sampled_task(self) -> Optional[NavToPartnerTask]: method close (line 1007) | def close(self) -> None: method all_observation_spaces_equal (line 1012) | def all_observation_spaces_equal(self) -> bool: method sample_scene (line 1022) | def sample_scene(self, force_advance_scene: bool): method next_task (line 1063) | def next_task( method reset (line 1129) | def reset(self): method set_seed (line 1136) | def set_seed(self, seed: int): FILE: allenact_plugins/robothor_plugin/robothor_tasks.py function spl_metric (line 24) | def spl_metric( class PointNavTask (line 41) | class PointNavTask(Task[RoboThorEnvironment]): method __init__ (line 44) | def __init__( method action_space (line 76) | def action_space(self): method reached_terminal_state (line 79) | def reached_terminal_state(self) -> bool: method class_action_names (line 83) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 86) | def close(self) -> None: method _step (line 89) | def _step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method render (line 117) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method _is_goal_in_range (line 124) | def _is_goal_in_range(self) -> Optional[bool]: method shaping (line 140) | def shaping(self) -> float: method judge (line 158) | def judge(self) -> float: method dist_to_target (line 177) | def dist_to_target(self): method metrics (line 180) | def metrics(self) -> Dict[str, Any]: class ObjectNavTask (line 207) | class ObjectNavTask(Task[RoboThorEnvironment]): method __init__ (line 210) | def __init__( method action_space (line 252) | def action_space(self): method reached_terminal_state (line 255) | def reached_terminal_state(self) -> bool: method class_action_names (line 259) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 262) | def close(self) -> None: method _step (line 265) | def _step(self, action: Union[int, Sequence[int]]) -> RLStepResult: method render (line 301) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method _is_goal_in_range (line 315) | def _is_goal_in_range(self) -> bool: method shaping (line 321) | def shaping(self) -> float: method judge (line 359) | def judge(self) -> float: method get_observations (line 376) | def get_observations(self, **kwargs) -> Any: method metrics (line 389) | def metrics(self) -> Dict[str, Any]: method query_expert (line 412) | def query_expert(self, end_action_only: bool = False, **kwargs) -> Tup... class NavToPartnerTask (line 453) | class NavToPartnerTask(Task[RoboThorEnvironment]): method __init__ (line 456) | def __init__( method action_space (line 486) | def action_space(self): method reached_terminal_state (line 494) | def reached_terminal_state(self) -> bool: method class_action_names (line 500) | def class_action_names(cls, **kwargs) -> Tuple[str, ...]: method close (line 503) | def close(self) -> None: method _step (line 506) | def _step(self, action: Tuple[int, int]) -> RLStepResult: method render (line 542) | def render(self, mode: str = "rgb", *args, **kwargs) -> np.ndarray: method judge (line 549) | def judge(self) -> float: method metrics (line 558) | def metrics(self) -> Dict[str, Any]: FILE: allenact_plugins/robothor_plugin/robothor_viz.py class ThorPositionTo2DFrameTranslator (line 23) | class ThorPositionTo2DFrameTranslator(object): method __init__ (line 24) | def __init__( method __call__ (line 34) | def __call__(self, position: Sequence[float]): class ThorViz (line 50) | class ThorViz(TrajectoryViz): method __init__ (line 51) | def __init__( method init_top_down_render (line 96) | def init_top_down_render(self): method iterate_scenes (line 106) | def iterate_scenes( method cached_map_data_path (line 115) | def cached_map_data_path(self, roomname: str) -> str: method get_translator (line 118) | def get_translator(self) -> Dict[str, Any]: method cached_image_path (line 142) | def cached_image_path(self, roomname: str) -> str: method make_top_down_views (line 147) | def make_top_down_views(self) -> Dict[str, np.ndarray]: method crop_viz_image (line 158) | def crop_viz_image(self, viz_image: np.ndarray) -> np.ndarray: method make_controller (line 168) | def make_controller(self): method get_agent_map_data (line 181) | def get_agent_map_data(self): method position_to_tuple (line 193) | def position_to_tuple(position: Dict[str, float]) -> Tuple[float, floa... method add_lines_to_map (line 197) | def add_lines_to_map( method add_line_to_map (line 226) | def add_line_to_map( method add_agent_view_triangle (line 254) | def add_agent_view_triangle( method visualize_agent_path (line 288) | def visualize_agent_path( method dump_top_down_view (line 342) | def dump_top_down_view(self, room_name: str, image_path: str): method make_fig (line 354) | def make_fig(self, episode: Any, episode_id: str) -> Figure: class ThorMultiViz (line 382) | class ThorMultiViz(ThorViz): method __init__ (line 383) | def __init__( method make_fig (line 400) | def make_fig(self, episode: Any, episode_id: str) -> Figure: FILE: allenact_plugins/robothor_plugin/scripts/make_objectnav_debug_dataset.py function create_debug_dataset_from_train_dataset (line 11) | def create_debug_dataset_from_train_dataset( FILE: allenact_plugins/setup.py function parse_req_file (line 8) | def parse_req_file(fname, initial=None): function get_version (line 45) | def get_version(fname): function run_setup (line 60) | def run_setup(): FILE: datasets/.habitat_downloader_helper.py function get_habitat_download_info (line 14) | def get_habitat_download_info(allow_create: bool = False): FILE: projects/babyai_baselines/experiments/base.py class BaseBabyAIExperimentConfig (line 32) | class BaseBabyAIExperimentConfig(ExperimentConfig, ABC): method METRIC_ACCUMULATE_INTERVAL (line 55) | def METRIC_ACCUMULATE_INTERVAL(cls): method get_sensors (line 59) | def get_sensors(cls) -> Sequence[Sensor]: method rl_loss_default (line 85) | def rl_loss_default(cls, alg: str, steps: Optional[int] = None): method _training_pipeline (line 113) | def _training_pipeline( method machine_params (line 156) | def machine_params( method create_model (line 181) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 192) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 195) | def train_task_sampler_args( method valid_task_sampler_args (line 209) | def valid_task_sampler_args( method test_task_sampler_args (line 219) | def test_task_sampler_args( FILE: projects/babyai_baselines/experiments/go_to_local/a2c.py class A2CBabyAIGoToLocalExperimentConfig (line 9) | class A2CBabyAIGoToLocalExperimentConfig(BaseBabyAIGoToLocalExperimentCo... method tag (line 22) | def tag(cls): method training_pipeline (line 26) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_local/base.py class BaseBabyAIGoToLocalExperimentConfig (line 24) | class BaseBabyAIGoToLocalExperimentConfig(BaseBabyAIExperimentConfig, ABC): method METRIC_ACCUMULATE_INTERVAL (line 47) | def METRIC_ACCUMULATE_INTERVAL(cls): method _training_pipeline (line 51) | def _training_pipeline( # type:ignore method create_model (line 100) | def create_model(cls, **kwargs) -> nn.Module: method valid_task_sampler_args (line 114) | def valid_task_sampler_args( FILE: projects/babyai_baselines/experiments/go_to_local/bc.py class PPOBabyAIGoToLocalExperimentConfig (line 7) | class PPOBabyAIGoToLocalExperimentConfig(BaseBabyAIGoToLocalExperimentCo... method tag (line 13) | def tag(cls): method training_pipeline (line 17) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_local/bc_teacher_forcing.py class BCTeacherForcingBabyAIGoToLocalExperimentConfig (line 9) | class BCTeacherForcingBabyAIGoToLocalExperimentConfig( method METRIC_ACCUMULATE_INTERVAL (line 19) | def METRIC_ACCUMULATE_INTERVAL(cls): method tag (line 23) | def tag(cls): method training_pipeline (line 27) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_local/dagger.py class DaggerBabyAIGoToLocalExperimentConfig (line 7) | class DaggerBabyAIGoToLocalExperimentConfig(BaseBabyAIGoToLocalExperimen... method tag (line 16) | def tag(cls): method training_pipeline (line 20) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_local/distributed_bc_offpolicy.py class DistributedBCOffPolicyBabyAIGoToLocalExperimentConfig (line 25) | class DistributedBCOffPolicyBabyAIGoToLocalExperimentConfig( method tag (line 31) | def tag(cls): method machine_params (line 35) | def machine_params( method expert_ce_loss_kwargs_generator (line 48) | def expert_ce_loss_kwargs_generator( method training_pipeline (line 54) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_local/distributed_bc_teacher_forcing.py class DistributedBCTeacherForcingBabyAIGoToLocalExperimentConfig (line 6) | class DistributedBCTeacherForcingBabyAIGoToLocalExperimentConfig( method METRIC_ACCUMULATE_INTERVAL (line 16) | def METRIC_ACCUMULATE_INTERVAL(cls): method tag (line 20) | def tag(cls): method machine_params (line 24) | def machine_params( FILE: projects/babyai_baselines/experiments/go_to_local/ppo.py class PPOBabyAIGoToLocalExperimentConfig (line 9) | class PPOBabyAIGoToLocalExperimentConfig(BaseBabyAIGoToLocalExperimentCo... method tag (line 22) | def tag(cls): method training_pipeline (line 26) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_obj/a2c.py class A2CBabyAIGoToObjExperimentConfig (line 7) | class A2CBabyAIGoToObjExperimentConfig(BaseBabyAIGoToObjExperimentConfig): method tag (line 13) | def tag(cls): method training_pipeline (line 17) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_obj/base.py class BaseBabyAIGoToObjExperimentConfig (line 22) | class BaseBabyAIGoToObjExperimentConfig(BaseBabyAIExperimentConfig, ABC): method METRIC_ACCUMULATE_INTERVAL (line 44) | def METRIC_ACCUMULATE_INTERVAL(cls): method _training_pipeline (line 48) | def _training_pipeline( # type:ignore method create_model (line 97) | def create_model(cls, **kwargs) -> nn.Module: FILE: projects/babyai_baselines/experiments/go_to_obj/bc.py class PPOBabyAIGoToObjExperimentConfig (line 7) | class PPOBabyAIGoToObjExperimentConfig(BaseBabyAIGoToObjExperimentConfig): method tag (line 13) | def tag(cls): method training_pipeline (line 17) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_obj/bc_teacher_forcing.py class PPOBabyAIGoToObjExperimentConfig (line 7) | class PPOBabyAIGoToObjExperimentConfig(BaseBabyAIGoToObjExperimentConfig): method tag (line 13) | def tag(cls): method training_pipeline (line 17) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_obj/dagger.py class DaggerBabyAIGoToObjExperimentConfig (line 7) | class DaggerBabyAIGoToObjExperimentConfig(BaseBabyAIGoToObjExperimentCon... method tag (line 16) | def tag(cls): method training_pipeline (line 20) | def training_pipeline(cls, **kwargs): FILE: projects/babyai_baselines/experiments/go_to_obj/ppo.py class PPOBabyAIGoToObjExperimentConfig (line 7) | class PPOBabyAIGoToObjExperimentConfig(BaseBabyAIGoToObjExperimentConfig): method tag (line 11) | def tag(cls): method training_pipeline (line 15) | def training_pipeline(cls, **kwargs): FILE: projects/gym_baselines/experiments/gym_base.py class GymBaseConfig (line 8) | class GymBaseConfig(ExperimentConfig, ABC): method _get_sampler_args (line 12) | def _get_sampler_args( method train_task_sampler_args (line 17) | def train_task_sampler_args( method valid_task_sampler_args (line 29) | def valid_task_sampler_args( method test_task_sampler_args (line 41) | def test_task_sampler_args( FILE: projects/gym_baselines/experiments/gym_humanoid_base.py class GymHumanoidBaseConfig (line 9) | class GymHumanoidBaseConfig(GymBaseConfig, ABC): method machine_params (line 11) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: FILE: projects/gym_baselines/experiments/gym_humanoid_ddppo.py class GymHumanoidPPOConfig (line 19) | class GymHumanoidPPOConfig(GymHumanoidBaseConfig, ABC): method training_pipeline (line 21) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: FILE: projects/gym_baselines/experiments/gym_mujoco_base.py class GymMoJoCoBaseConfig (line 9) | class GymMoJoCoBaseConfig(GymBaseConfig, ABC): method machine_params (line 11) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: FILE: projects/gym_baselines/experiments/gym_mujoco_ddppo.py class GymMuJoCoPPOConfig (line 19) | class GymMuJoCoPPOConfig(GymMoJoCoBaseConfig, ABC): method training_pipeline (line 21) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_ant_ddppo.py class GymMuJoCoAntConfig (line 16) | class GymMuJoCoAntConfig(GymMuJoCoPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 42) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 45) | def _get_sampler_args( method tag (line 83) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_halfcheetah_ddppo.py class GymMuJoCoHalfCheetahConfig (line 16) | class GymMuJoCoHalfCheetahConfig(GymMuJoCoPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 42) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 45) | def _get_sampler_args( method tag (line 83) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_hopper_ddppo.py class GymMuJoCoHopperConfig (line 16) | class GymMuJoCoHopperConfig(GymMuJoCoPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 42) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 45) | def _get_sampler_args( method tag (line 83) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_humanoid_ddppo.py class GymMuJoCoHumanoidConfig (line 16) | class GymMuJoCoHumanoidConfig(GymHumanoidPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 44) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 47) | def _get_sampler_args( method tag (line 85) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_inverteddoublependulum_ddppo.py class GymMuJoInvertedDoublePendulumConfig (line 16) | class GymMuJoInvertedDoublePendulumConfig(GymMuJoCoPPOConfig): method create_model (line 25) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 44) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 47) | def _get_sampler_args( method tag (line 85) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_invertedpendulum_ddppo.py class GymMuJoCoInvertedPendulumConfig (line 16) | class GymMuJoCoInvertedPendulumConfig(GymMuJoCoPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 42) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 45) | def _get_sampler_args( method tag (line 83) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_reacher_ddppo.py class GymMuJoCoReacherConfig (line 16) | class GymMuJoCoReacherConfig(GymMuJoCoPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 42) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 45) | def _get_sampler_args( method tag (line 83) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_swimmer_ddppo.py class GymMuJoCoSwimmerConfig (line 16) | class GymMuJoCoSwimmerConfig(GymMuJoCoPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 42) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 45) | def _get_sampler_args( method tag (line 83) | def tag(cls) -> str: FILE: projects/gym_baselines/experiments/mujoco/gym_mujoco_walker2d_ddppo.py class GymMuJoCoWalkerConfig (line 16) | class GymMuJoCoWalkerConfig(GymMuJoCoPPOConfig): method create_model (line 23) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 42) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _get_sampler_args (line 45) | def _get_sampler_args( method tag (line 83) | def tag(cls) -> str: FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/armpointnav_base.py class ArmPointNavBaseConfig (line 10) | class ArmPointNavBaseConfig(ExperimentConfig, ABC): method __init__ (line 26) | def __init__(self): method preprocessors (line 37) | def preprocessors(cls) -> Sequence[Union[Preprocessor, Builder[Preproc... FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/armpointnav_mixin_ddppo.py class ArmPointNavMixInPPOConfig (line 17) | class ArmPointNavMixInPPOConfig(ArmPointNavBaseConfig): method training_pipeline (line 18) | def training_pipeline(self, **kwargs): FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/armpointnav_mixin_simplegru.py class ArmPointNavMixInSimpleGRUConfig (line 17) | class ArmPointNavMixInSimpleGRUConfig(ArmPointNavBaseConfig): method preprocessors (line 21) | def preprocessors(cls) -> Sequence[Union[Preprocessor, Builder[Preproc... method create_model (line 26) | def create_model(cls, **kwargs) -> nn.Module: FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/armpointnav_thor_base.py class ArmPointNavThorBaseConfig (line 28) | class ArmPointNavThorBaseConfig(ArmPointNavBaseConfig, ABC): method __init__ (line 57) | def __init__(self): method machine_params (line 68) | def machine_params(self, mode="train", **kwargs): method make_sampler_fn (line 119) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _partition_inds (line 136) | def _partition_inds(n: int, num_parts: int): method _get_sampler_args_for_scene_split (line 141) | def _get_sampler_args_for_scene_split( method train_task_sampler_args (line 178) | def train_task_sampler_args( method valid_task_sampler_args (line 203) | def valid_task_sampler_args( method test_task_sampler_args (line 229) | def test_task_sampler_args( FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/ithor/armpointnav_depth.py class ArmPointNavDepth (line 22) | class ArmPointNavDepth( method __init__ (line 45) | def __init__(self): method tag (line 57) | def tag(cls): FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/ithor/armpointnav_disjoint_depth.py class ArmPointNavDisjointDepth (line 16) | class ArmPointNavDisjointDepth(ArmPointNavDepth): method __init__ (line 22) | def __init__(self): method create_model (line 34) | def create_model(cls, **kwargs) -> nn.Module: method tag (line 44) | def tag(cls): FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/ithor/armpointnav_ithor_base.py class ArmPointNaviThorBaseConfig (line 12) | class ArmPointNaviThorBaseConfig(ArmPointNavThorBaseConfig, ABC): FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/ithor/armpointnav_no_vision.py class ArmPointNavNoVision (line 22) | class ArmPointNavNoVision( method __init__ (line 45) | def __init__(self): method tag (line 57) | def tag(cls): FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/ithor/armpointnav_rgb.py class ArmPointNavRGB (line 22) | class ArmPointNavRGB( method __init__ (line 45) | def __init__(self): method tag (line 57) | def tag(cls): FILE: projects/manipulathor_baselines/armpointnav_baselines/experiments/ithor/armpointnav_rgbdepth.py class ArmPointNavRGBDepth (line 23) | class ArmPointNavRGBDepth( method __init__ (line 52) | def __init__(self): method tag (line 64) | def tag(cls): FILE: projects/manipulathor_baselines/armpointnav_baselines/models/arm_pointnav_models.py class ArmPointNavBaselineActorCritic (line 28) | class ArmPointNavBaselineActorCritic(ActorCriticModel[CategoricalDistr]): method __init__ (line 41) | def __init__( method recurrent_hidden_state_size (line 95) | def recurrent_hidden_state_size(self) -> int: method num_recurrent_layers (line 100) | def num_recurrent_layers(self) -> int: method _recurrent_memory_specification (line 104) | def _recurrent_memory_specification(self): method get_relative_distance_embedding (line 116) | def get_relative_distance_embedding( method forward (line 122) | def forward( # type:ignore FILE: projects/manipulathor_baselines/armpointnav_baselines/models/base_models.py class LinearActorHeadNoCategory (line 5) | class LinearActorHeadNoCategory(nn.Module): method __init__ (line 6) | def __init__(self, num_inputs: int, num_outputs: int): method forward (line 13) | def forward(self, x: torch.FloatTensor): # type: ignore FILE: projects/manipulathor_baselines/armpointnav_baselines/models/disjoint_arm_pointnav_models.py class DisjointArmPointNavBaselineActorCritic (line 30) | class DisjointArmPointNavBaselineActorCritic(ActorCriticModel[Categorica... method __init__ (line 46) | def __init__( method recurrent_hidden_state_size (line 104) | def recurrent_hidden_state_size(self) -> int: method num_recurrent_layers (line 109) | def num_recurrent_layers(self) -> int: method _recurrent_memory_specification (line 113) | def _recurrent_memory_specification(self): method forward (line 125) | def forward( # type:ignore FILE: projects/manipulathor_baselines/armpointnav_baselines/models/manipulathor_net_utils.py function upshuffle (line 7) | def upshuffle( function upshufflenorelu (line 23) | def upshufflenorelu( function combine_block_w_bn (line 38) | def combine_block_w_bn(in_planes, out_planes): function conv2d_block (line 46) | def conv2d_block(in_planes, out_planes, kernel_size, stride=1, padding=1): function combine_block_w_do (line 56) | def combine_block_w_do(in_planes, out_planes, dropout=0.0): function combine_block_no_do (line 64) | def combine_block_no_do(in_planes, out_planes): function linear_block (line 71) | def linear_block(in_features, out_features, dropout=0.0): function linear_block_norelu (line 79) | def linear_block_norelu(in_features, out_features): function input_embedding_net (line 85) | def input_embedding_net(list_of_feature_sizes, dropout=0.0): function _upsample_add (line 96) | def _upsample_add(x, y): function replace_all_relu_w_leakyrelu (line 101) | def replace_all_relu_w_leakyrelu(model): function replace_all_leakyrelu_w_relu (line 114) | def replace_all_leakyrelu_w_relu(model): function replace_all_bn_w_groupnorm (line 125) | def replace_all_bn_w_groupnorm(model): function flat_temporal (line 141) | def flat_temporal(tensor, batch_size, sequence_length): function unflat_temporal (line 148) | def unflat_temporal(tensor, batch_size, sequence_length): FILE: projects/objectnav_baselines/experiments/clip/mixins.py class LookDownFirstResnetTensorNavActorCritic (line 27) | class LookDownFirstResnetTensorNavActorCritic(ResnetTensorNavActorCritic): method __init__ (line 28) | def __init__(self, look_down_action_index: int, **kwargs): method forward (line 37) | def forward( # type:ignore class ClipResNetPreprocessGRUActorCriticMixin (line 61) | class ClipResNetPreprocessGRUActorCriticMixin: method preprocessors (line 68) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 114) | def create_model( FILE: projects/objectnav_baselines/experiments/habitat/clip/objectnav_habitat_rgb_clipresnet50gru_ddppo.py class ObjectNavHabitatRGBClipResNet50GRUDDPPOExperimentConfig (line 22) | class ObjectNavHabitatRGBClipResNet50GRUDDPPOExperimentConfig( method __init__ (line 29) | def __init__(self, lr: float, **kwargs): method SENSORS (line 42) | def SENSORS(self): method training_pipeline (line 54) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 62) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 65) | def create_model(self, **kwargs) -> nn.Module: method tag (line 73) | def tag(self): FILE: projects/objectnav_baselines/experiments/habitat/clip/objectnav_habitat_rgb_clipresnet50gru_ddppo_increasingrollouts.py class ObjectNavHabitatRGBClipResNet50GRUDDPPOIncreasingLengthExpConfig (line 18) | class ObjectNavHabitatRGBClipResNet50GRUDDPPOIncreasingLengthExpConfig( method __init__ (line 21) | def __init__(self, lr=1e-4, **kwargs): method training_pipeline (line 25) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method tag (line 101) | def tag(self): FILE: projects/objectnav_baselines/experiments/habitat/objectnav_habitat_base.py function create_objectnav_config (line 39) | def create_objectnav_config( class ObjectNavHabitatBaseConfig (line 111) | class ObjectNavHabitatBaseConfig(ObjectNavBaseConfig, ABC): method __init__ (line 145) | def __init__( method _create_config (line 193) | def _create_config( method DEFAULT_OBJECT_CATEGORIES_TO_IND (line 221) | def DEFAULT_OBJECT_CATEGORIES_TO_IND(self): method TASK_DATA_DIR_TEMPLATE (line 259) | def TASK_DATA_DIR_TEMPLATE(self): method BASE_CONFIG_YAML_PATH (line 265) | def BASE_CONFIG_YAML_PATH(self): method TRAIN_CONFIG (line 271) | def TRAIN_CONFIG(self): method VALID_CONFIG (line 281) | def VALID_CONFIG(self): method TEST_CONFIG (line 292) | def TEST_CONFIG(self): method TRAIN_CONFIGS_PER_PROCESS (line 302) | def TRAIN_CONFIGS_PER_PROCESS(self): method TEST_CONFIG_PER_PROCESS (line 363) | def TEST_CONFIG_PER_PROCESS(self): method train_scenes_path (line 366) | def train_scenes_path(self): method valid_scenes_path (line 369) | def valid_scenes_path(self): method test_scenes_path (line 372) | def test_scenes_path(self): method tag (line 377) | def tag(self): method preprocessors (line 387) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method machine_params (line 390) | def machine_params(self, mode="train", **kwargs): method make_sampler_fn (line 429) | def make_sampler_fn(self, **kwargs) -> TaskSampler: method train_task_sampler_args (line 437) | def train_task_sampler_args( method valid_task_sampler_args (line 453) | def valid_task_sampler_args( method test_task_sampler_args (line 474) | def test_task_sampler_args( FILE: projects/objectnav_baselines/experiments/ithor/objectnav_ithor_base.py class ObjectNaviThorBaseConfig (line 11) | class ObjectNaviThorBaseConfig(ObjectNavThorBaseConfig, ABC): FILE: projects/objectnav_baselines/experiments/ithor/objectnav_ithor_depth_resnet18gru_ddppo.py class ObjectNaviThorDepthPPOExperimentConfig (line 18) | class ObjectNaviThorDepthPPOExperimentConfig(ObjectNaviThorBaseConfig): method __init__ (line 34) | def __init__(self, **kwargs): method training_pipeline (line 44) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 51) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 54) | def create_model(self, **kwargs) -> nn.Module: method tag (line 59) | def tag(self): FILE: projects/objectnav_baselines/experiments/ithor/objectnav_ithor_rgb_resnet18gru_ddppo.py class ObjectNaviThorRGBPPOExperimentConfig (line 20) | class ObjectNaviThorRGBPPOExperimentConfig(ObjectNaviThorBaseConfig): method __init__ (line 36) | def __init__(self, **kwargs): method training_pipeline (line 46) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 53) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 56) | def create_model(self, **kwargs) -> nn.Module: method tag (line 62) | def tag(cls): FILE: projects/objectnav_baselines/experiments/ithor/objectnav_ithor_rgbd_resnet18gru_ddppo.py class ObjectNaviThorRGBDPPOExperimentConfig (line 21) | class ObjectNaviThorRGBDPPOExperimentConfig(ObjectNaviThorBaseConfig): method __init__ (line 43) | def __init__(self, **kwargs): method training_pipeline (line 53) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 60) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 63) | def create_model(self, **kwargs) -> nn.Module: method tag (line 68) | def tag(self): FILE: projects/objectnav_baselines/experiments/objectnav_base.py class ObjectNavBaseConfig (line 10) | class ObjectNavBaseConfig(ExperimentConfig, ABC): method __init__ (line 27) | def __init__(self): method preprocessors (line 35) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... FILE: projects/objectnav_baselines/experiments/objectnav_thor_base.py class ObjectNavThorBaseConfig (line 45) | class ObjectNavThorBaseConfig(ObjectNavBaseConfig, ABC): method __init__ (line 66) | def __init__( method env_args (line 98) | def env_args(self): method machine_params (line 124) | def machine_params(self, mode="train", **kwargs): method make_sampler_fn (line 184) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _partition_inds (line 188) | def _partition_inds(n: int, num_parts: int): method _get_sampler_args_for_scene_split (line 193) | def _get_sampler_args_for_scene_split( method train_task_sampler_args (line 276) | def train_task_sampler_args( method valid_task_sampler_args (line 299) | def valid_task_sampler_args( method test_task_sampler_args (line 321) | def test_task_sampler_args( FILE: projects/objectnav_baselines/experiments/robothor/beta/objectnav_robothor_rgb_resnetgru_ddppo_and_gbc.py class ObjectNavRoboThorResNet18GRURGBPPOExperimentConfig (line 28) | class ObjectNavRoboThorResNet18GRURGBPPOExperimentConfig(ObjectNavRoboTh... method __init__ (line 47) | def __init__(self, **kwargs): method preprocessors (line 57) | def preprocessors(self): method create_model (line 60) | def create_model(self, **kwargs): method training_pipeline (line 65) | def training_pipeline(self, **kwargs): method tag (line 114) | def tag(self): FILE: projects/objectnav_baselines/experiments/robothor/beta/objectnav_robothor_rgb_unfrozenresnet18gru_vdr_ddppo.py function compute_inv_dyn_action_logits (line 43) | def compute_inv_dyn_action_logits( class LastActionSuccessSensor (line 54) | class LastActionSuccessSensor( method __init__ (line 60) | def __init__(self, uuid: str = "last_action_success", **kwargs: Any): method get_observation (line 65) | def get_observation( class VisibleObjectTypesSensor (line 75) | class VisibleObjectTypesSensor( method __init__ (line 81) | def __init__(self, uuid: str = "visible_objects", **kwargs: Any): method get_observation (line 93) | def get_observation( class ObjectNavRoboThorVdrTmpRGBExperimentConfig (line 107) | class ObjectNavRoboThorVdrTmpRGBExperimentConfig(ObjectNavRoboThorBaseCo... method __init__ (line 122) | def __init__(self, **kwargs): method training_pipeline (line 134) | def training_pipeline(self, **kwargs): method create_model (line 233) | def create_model(self, **kwargs) -> nn.Module: method tag (line 242) | def tag(self): FILE: projects/objectnav_baselines/experiments/robothor/clip/objectnav_robothor_rgb_clipresnet50gru_ddppo.py class ObjectNavRoboThorClipRGBPPOExperimentConfig (line 21) | class ObjectNavRoboThorClipRGBPPOExperimentConfig(ObjectNavRoboThorBaseC... method __init__ (line 41) | def __init__(self, add_prev_actions: bool = False, **kwargs): method training_pipeline (line 52) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 59) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 62) | def create_model(self, **kwargs) -> nn.Module: method tag (line 70) | def tag(cls): FILE: projects/objectnav_baselines/experiments/robothor/clip/objectnav_robothor_rgb_clipresnet50x16gru_ddppo.py class ObjectNavRoboThorRGBPPOExperimentConfig (line 21) | class ObjectNavRoboThorRGBPPOExperimentConfig(ObjectNavRoboThorBaseConfig): method __init__ (line 41) | def __init__(self, add_prev_actions: bool = False, **kwargs): method training_pipeline (line 52) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 59) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 62) | def create_model(self, **kwargs) -> nn.Module: method tag (line 70) | def tag(cls): FILE: projects/objectnav_baselines/experiments/robothor/objectnav_robothor_base.py class ObjectNavRoboThorBaseConfig (line 13) | class ObjectNavRoboThorBaseConfig(ObjectNavThorBaseConfig, ABC): method train_task_sampler_args (line 46) | def train_task_sampler_args( FILE: projects/objectnav_baselines/experiments/robothor/objectnav_robothor_depth_resnet18gru_ddppo.py class ObjectNavRoboThorRGBPPOExperimentConfig (line 18) | class ObjectNavRoboThorRGBPPOExperimentConfig(ObjectNavRoboThorBaseConfig): method __init__ (line 34) | def __init__(self, **kwargs): method training_pipeline (line 44) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 51) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 54) | def create_model(self, **kwargs) -> nn.Module: method tag (line 59) | def tag(self): FILE: projects/objectnav_baselines/experiments/robothor/objectnav_robothor_rgb_resnet18gru_dagger.py class ObjectNavRoboThorRGBDAggerExperimentConfig (line 22) | class ObjectNavRoboThorRGBDAggerExperimentConfig(ObjectNavRoboThorBaseCo... method __init__ (line 41) | def __init__(self, **kwargs): method training_pipeline (line 53) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 58) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 61) | def create_model(self, **kwargs) -> nn.Module: method tag (line 67) | def tag(cls): FILE: projects/objectnav_baselines/experiments/robothor/objectnav_robothor_rgb_resnet18gru_ddppo.py class ObjectNavRoboThorRGBPPOExperimentConfig (line 20) | class ObjectNavRoboThorRGBPPOExperimentConfig(ObjectNavRoboThorBaseConfig): method __init__ (line 36) | def __init__(self, **kwargs): method training_pipeline (line 46) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 53) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 56) | def create_model(self, **kwargs) -> nn.Module: method tag (line 62) | def tag(cls): FILE: projects/objectnav_baselines/experiments/robothor/objectnav_robothor_rgb_resnet50gru_ddppo.py class ObjectNavRoboThorRGBPPOExperimentConfig (line 20) | class ObjectNavRoboThorRGBPPOExperimentConfig(ObjectNavRoboThorBaseConfig): method __init__ (line 36) | def __init__(self, **kwargs): method training_pipeline (line 46) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 53) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 56) | def create_model(self, **kwargs) -> nn.Module: method tag (line 61) | def tag(self): FILE: projects/objectnav_baselines/experiments/robothor/objectnav_robothor_rgb_unfrozenresnet18gru_ddppo.py class ObjectNavRoboThorRGBPPOExperimentConfig (line 17) | class ObjectNavRoboThorRGBPPOExperimentConfig(ObjectNavRoboThorBaseConfig): method __init__ (line 33) | def __init__(self, **kwargs): method training_pipeline (line 45) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 52) | def create_model(self, **kwargs) -> nn.Module: method tag (line 55) | def tag(self): FILE: projects/objectnav_baselines/experiments/robothor/objectnav_robothor_rgbd_resnet18gru_ddppo.py class ObjectNavRoboThorRGBPPOExperimentConfig (line 21) | class ObjectNavRoboThorRGBPPOExperimentConfig(ObjectNavRoboThorBaseConfig): method __init__ (line 43) | def __init__(self, **kwargs): method training_pipeline (line 53) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 60) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 63) | def create_model(self, **kwargs) -> nn.Module: method tag (line 68) | def tag(self): FILE: projects/objectnav_baselines/mixins.py class ResNetPreprocessGRUActorCriticMixin (line 51) | class ResNetPreprocessGRUActorCriticMixin: method preprocessors (line 57) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 110) | def create_model(self, **kwargs) -> nn.Module: class ObjectNavUnfrozenResNetWithGRUActorCriticMixin (line 132) | class ObjectNavUnfrozenResNetWithGRUActorCriticMixin: method create_model (line 140) | def create_model(self, **kwargs) -> nn.Module: class ObjectNavDAggerMixin (line 173) | class ObjectNavDAggerMixin: method training_pipeline (line 175) | def training_pipeline( function update_with_auxiliary_losses (line 236) | def update_with_auxiliary_losses( class ObjectNavPPOMixin (line 332) | class ObjectNavPPOMixin: method training_pipeline (line 334) | def training_pipeline( FILE: projects/pointnav_baselines/experiments/habitat/clip/pointnav_habitat_rgb_clipresnet50gru_ddppo.py class PointNavHabitatRGBClipResNet50GRUDDPPOExperimentConfig (line 21) | class PointNavHabitatRGBClipResNet50GRUDDPPOExperimentConfig(PointNavHab... method __init__ (line 38) | def __init__(self, add_prev_actions: bool = False, **kwargs): method training_pipeline (line 49) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method preprocessors (line 57) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method create_model (line 60) | def create_model(self, **kwargs) -> nn.Module: method tag (line 68) | def tag(cls): FILE: projects/pointnav_baselines/experiments/habitat/pointnav_habitat_base.py function create_pointnav_config (line 34) | def create_pointnav_config( class PointNavHabitatBaseConfig (line 99) | class PointNavHabitatBaseConfig(PointNavBaseConfig, ABC): method __init__ (line 141) | def __init__( method train_scenes_path (line 236) | def train_scenes_path(self): method valid_scenes_path (line 239) | def valid_scenes_path(self): method test_scenes_path (line 242) | def test_scenes_path(self): method tag (line 248) | def tag(cls): method preprocessors (line 251) | def preprocessors(self) -> Sequence[Union[Preprocessor, Builder[Prepro... method machine_params (line 254) | def machine_params(self, mode="train", **kwargs): method make_sampler_fn (line 294) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 299) | def train_task_sampler_args( method valid_task_sampler_args (line 316) | def valid_task_sampler_args( method test_task_sampler_args (line 336) | def test_task_sampler_args( FILE: projects/pointnav_baselines/experiments/habitat/pointnav_habitat_depth_simpleconvgru_ddppo.py class PointNavHabitatDepthDeterministiSimpleConvGRUDDPPOExperimentConfig (line 15) | class PointNavHabitatDepthDeterministiSimpleConvGRUDDPPOExperimentConfig( method __init__ (line 30) | def __init__(self): method training_pipeline (line 42) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 50) | def create_model(self, **kwargs): method tag (line 53) | def tag(self): FILE: projects/pointnav_baselines/experiments/habitat/pointnav_habitat_rgb_simpleconvgru_ddppo.py class PointNavHabitatDepthDeterministiSimpleConvGRUDDPPOExperimentConfig (line 15) | class PointNavHabitatDepthDeterministiSimpleConvGRUDDPPOExperimentConfig( method __init__ (line 30) | def __init__(self): method training_pipeline (line 42) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 50) | def create_model(self, **kwargs): method tag (line 54) | def tag(cls): FILE: projects/pointnav_baselines/experiments/habitat/pointnav_habitat_rgbd_simpleconvgru_ddppo.py class PointNavHabitatDepthDeterministiSimpleConvGRUDDPPOExperimentConfig (line 16) | class PointNavHabitatDepthDeterministiSimpleConvGRUDDPPOExperimentConfig( method __init__ (line 36) | def __init__(self): method training_pipeline (line 48) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 56) | def create_model(self, **kwargs): method tag (line 59) | def tag(self): FILE: projects/pointnav_baselines/experiments/ithor/pointnav_ithor_base.py class PointNaviThorBaseConfig (line 9) | class PointNaviThorBaseConfig(PointNavThorBaseConfig, ABC): FILE: projects/pointnav_baselines/experiments/ithor/pointnav_ithor_depth_simpleconvgru_ddppo.py class PointNaviThorDepthPPOExperimentConfig (line 15) | class PointNaviThorDepthPPOExperimentConfig(PointNaviThorBaseConfig): method __init__ (line 29) | def __init__(self): method training_pipeline (line 41) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 49) | def create_model(self, **kwargs): method tag (line 52) | def tag(self): FILE: projects/pointnav_baselines/experiments/ithor/pointnav_ithor_rgb_simpleconvgru_ddppo.py class PointNaviThorRGBPPOExperimentConfig (line 13) | class PointNaviThorRGBPPOExperimentConfig(PointNaviThorBaseConfig): method __init__ (line 26) | def __init__(self): method training_pipeline (line 38) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 46) | def create_model(self, **kwargs): method tag (line 49) | def tag(self): FILE: projects/pointnav_baselines/experiments/ithor/pointnav_ithor_rgbd_simpleconvgru_ddppo.py class PointNaviThorRGBDPPOExperimentConfig (line 14) | class PointNaviThorRGBDPPOExperimentConfig(PointNaviThorBaseConfig): method __init__ (line 34) | def __init__(self): method training_pipeline (line 46) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 54) | def create_model(self, **kwargs): method tag (line 57) | def tag(self): FILE: projects/pointnav_baselines/experiments/pointnav_base.py class PointNavBaseConfig (line 8) | class PointNavBaseConfig(ExperimentConfig, ABC): method __init__ (line 24) | def __init__(self): FILE: projects/pointnav_baselines/experiments/pointnav_thor_base.py class PointNavThorBaseConfig (line 37) | class PointNavThorBaseConfig(PointNavBaseConfig, ABC): method __init__ (line 52) | def __init__(self): method preprocessors (line 67) | def preprocessors(self): method machine_params (line 70) | def machine_params(self, mode="train", **kwargs): method make_sampler_fn (line 117) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _partition_inds (line 121) | def _partition_inds(n: int, num_parts: int): method _get_sampler_args_for_scene_split (line 126) | def _get_sampler_args_for_scene_split( method train_task_sampler_args (line 195) | def train_task_sampler_args( method valid_task_sampler_args (line 216) | def valid_task_sampler_args( method test_task_sampler_args (line 237) | def test_task_sampler_args( FILE: projects/pointnav_baselines/experiments/robothor/pointnav_robothor_base.py class PointNavRoboThorBaseConfig (line 9) | class PointNavRoboThorBaseConfig(PointNavThorBaseConfig, ABC): FILE: projects/pointnav_baselines/experiments/robothor/pointnav_robothor_depth_simpleconvgru_ddppo.py class PointNavRoboThorRGBPPOExperimentConfig (line 17) | class PointNavRoboThorRGBPPOExperimentConfig( method __init__ (line 33) | def __init__(self): method training_pipeline (line 45) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 53) | def create_model(self, **kwargs): method tag (line 56) | def tag(self): FILE: projects/pointnav_baselines/experiments/robothor/pointnav_robothor_rgb_simpleconvgru_ddppo.py class PointNavRoboThorRGBPPOExperimentConfig (line 13) | class PointNavRoboThorRGBPPOExperimentConfig( method __init__ (line 29) | def __init__(self): method training_pipeline (line 41) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 49) | def create_model(self, **kwargs): method tag (line 52) | def tag(self): FILE: projects/pointnav_baselines/experiments/robothor/pointnav_robothor_rgbd_simpleconvgru_ddppo.py class PointNavRoboThorRGBPPOExperimentConfig (line 15) | class PointNavRoboThorRGBPPOExperimentConfig( method __init__ (line 37) | def __init__(self): method training_pipeline (line 49) | def training_pipeline(self, **kwargs) -> TrainingPipeline: method create_model (line 57) | def create_model(self, **kwargs): method tag (line 60) | def tag(self): FILE: projects/pointnav_baselines/mixins.py class TargetCoordinatesSensorHabitat (line 28) | class TargetCoordinatesSensorHabitat: #type:ignore class PointNavUnfrozenResNetWithGRUActorCriticMixin (line 38) | class PointNavUnfrozenResNetWithGRUActorCriticMixin: method create_model (line 46) | def create_model(self, **kwargs) -> nn.Module: class PointNavPPOMixin (line 90) | class PointNavPPOMixin: method training_pipeline (line 92) | def training_pipeline( FILE: projects/tutorials/distributed_objectnav_tutorial.py class DistributedObjectNavRoboThorRGBPPOExperimentConfig (line 75) | class DistributedObjectNavRoboThorRGBPPOExperimentConfig(BaseConfig): method tag (line 76) | def tag(self) -> str: method env_args (line 95) | def env_args(self): method __init__ (line 107) | def __init__( method machine_params (line 138) | def machine_params(self, mode="train", **kwargs): method lr_scheduler (line 199) | def lr_scheduler(small_batch_steps, transition_steps, ppo_steps, lr_sc... method training_pipeline (line 239) | def training_pipeline(self, **kwargs): FILE: projects/tutorials/gym_mujoco_tutorial.py class HandManipulateTutorialExperimentConfig (line 71) | class HandManipulateTutorialExperimentConfig(ExperimentConfig): method tag (line 73) | def tag(cls) -> str: method create_model (line 100) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 131) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 141) | def train_task_sampler_args( method valid_task_sampler_args (line 153) | def valid_task_sampler_args( method test_task_sampler_args (line 165) | def test_task_sampler_args( method _get_sampler_args (line 182) | def _get_sampler_args( method machine_params (line 233) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: method training_pipeline (line 263) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: FILE: projects/tutorials/gym_tutorial.py class GymTutorialExperimentConfig (line 73) | class GymTutorialExperimentConfig(ExperimentConfig): method tag (line 75) | def tag(cls) -> str: method create_model (line 102) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 124) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 134) | def train_task_sampler_args( method valid_task_sampler_args (line 146) | def valid_task_sampler_args( method test_task_sampler_args (line 158) | def test_task_sampler_args( method _get_sampler_args (line 175) | def _get_sampler_args( method machine_params (line 229) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: method training_pipeline (line 257) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: FILE: projects/tutorials/minigrid_offpolicy_tutorial.py class BCOffPolicyBabyAIGoToLocalExperimentConfig (line 124) | class BCOffPolicyBabyAIGoToLocalExperimentConfig(BaseBabyAIGoToLocalExpe... method tag (line 132) | def tag(cls): method METRIC_ACCUMULATE_INTERVAL (line 136) | def METRIC_ACCUMULATE_INTERVAL(cls): method training_pipeline (line 141) | def training_pipeline(cls, **kwargs): FILE: projects/tutorials/minigrid_tutorial.py class MiniGridTutorialExperimentConfig (line 100) | class MiniGridTutorialExperimentConfig(ExperimentConfig): method tag (line 107) | def tag(cls) -> str: method create_model (line 138) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 158) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 172) | def train_task_sampler_args( method valid_task_sampler_args (line 182) | def valid_task_sampler_args( method test_task_sampler_args (line 192) | def test_task_sampler_args( method _get_sampler_args (line 208) | def _get_sampler_args(self, process_ind: int, mode: str) -> Dict[str, ... method make_env (line 244) | def make_env(*args, **kwargs): method machine_params (line 267) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: method training_pipeline (line 286) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: FILE: projects/tutorials/minigrid_tutorial_conds.py class ConditionedLinearActorCriticHead (line 39) | class ConditionedLinearActorCriticHead(nn.Module): method __init__ (line 40) | def __init__( method lower_policy (line 54) | def lower_policy(self, *args, **kwargs): method forward (line 61) | def forward(self, x): class ConditionedLinearActorCritic (line 88) | class ConditionedLinearActorCritic(ActorCriticModel[SequentialDistr]): method __init__ (line 89) | def __init__( method _recurrent_memory_specification (line 116) | def _recurrent_memory_specification(self): method forward (line 119) | def forward(self, observations, memory, prev_actions, masks): class ConditionedRNNActorCritic (line 133) | class ConditionedRNNActorCritic(ActorCriticModel[SequentialDistr]): method __init__ (line 134) | def __init__( method recurrent_hidden_state_size (line 188) | def recurrent_hidden_state_size(self) -> int: method num_recurrent_layers (line 192) | def num_recurrent_layers(self) -> int: method _recurrent_memory_specification (line 195) | def _recurrent_memory_specification(self): method forward (line 207) | def forward( # type:ignore class ConditionedMiniGridSimpleConvRNN (line 235) | class ConditionedMiniGridSimpleConvRNN(MiniGridSimpleConvBase): method __init__ (line 236) | def __init__( method num_recurrent_layers (line 285) | def num_recurrent_layers(self): method recurrent_hidden_state_size (line 289) | def recurrent_hidden_state_size(self): method _recurrent_memory_specification (line 292) | def _recurrent_memory_specification(self): class ConditionedMiniGridTask (line 305) | class ConditionedMiniGridTask(MiniGridTask): method action_space (line 312) | def action_space(self) -> gym.spaces.Dict: method _step (line 317) | def _step(self, action: Dict[str, int]) -> RLStepResult: method query_expert (line 334) | def query_expert(self, **kwargs) -> Tuple[int, bool]: class MiniGridTutorialExperimentConfig (line 354) | class MiniGridTutorialExperimentConfig(ExperimentConfig): method tag (line 356) | def tag(cls) -> str: method create_model (line 369) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 381) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 384) | def train_task_sampler_args( method valid_task_sampler_args (line 394) | def valid_task_sampler_args( method test_task_sampler_args (line 404) | def test_task_sampler_args( method _get_sampler_args (line 414) | def _get_sampler_args(self, process_ind: int, mode: str) -> Dict[str, ... method make_env (line 453) | def make_env(*args, **kwargs): method machine_params (line 457) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: method training_pipeline (line 464) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: FILE: projects/tutorials/navtopartner_robothor_rgb_ppo.py class NavToPartnerRoboThorRGBPPOExperimentConfig (line 35) | class NavToPartnerRoboThorRGBPPOExperimentConfig(ExperimentConfig): method tag (line 82) | def tag(cls): method training_pipeline (line 86) | def training_pipeline(cls, **kwargs): method split_num_processes (line 119) | def split_num_processes(self, ndevices): method get_viz (line 130) | def get_viz(self, mode): method machine_params (line 150) | def machine_params(self, mode="train", **kwargs): method create_model (line 179) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 193) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _partition_inds (line 198) | def _partition_inds(n: int, num_parts: int): method _get_sampler_args_for_scene_split (line 203) | def _get_sampler_args_for_scene_split( method train_task_sampler_args (line 242) | def train_task_sampler_args( method valid_task_sampler_args (line 269) | def valid_task_sampler_args( method test_task_sampler_args (line 297) | def test_task_sampler_args( FILE: projects/tutorials/object_nav_ithor_dagger_then_ppo_one_object.py class ObjectNavThorDaggerThenPPOExperimentConfig (line 21) | class ObjectNavThorDaggerThenPPOExperimentConfig(ObjectNavThorPPOExperim... method tag (line 34) | def tag(cls): method training_pipeline (line 38) | def training_pipeline(cls, **kwargs): FILE: projects/tutorials/object_nav_ithor_dagger_then_ppo_one_object_viz.py class ObjectNavThorDaggerThenPPOVizExperimentConfig (line 14) | class ObjectNavThorDaggerThenPPOVizExperimentConfig( method tag (line 25) | def tag(cls): method get_viz (line 30) | def get_viz(self, mode): method machine_params (line 53) | def machine_params(self, mode="train", **kwargs): FILE: projects/tutorials/object_nav_ithor_ppo_one_object.py class ObjectNavThorPPOExperimentConfig (line 31) | class ObjectNavThorPPOExperimentConfig(ExperimentConfig): method tag (line 67) | def tag(cls): method training_pipeline (line 71) | def training_pipeline(cls, **kwargs): method machine_params (line 111) | def machine_params(cls, mode="train", **kwargs): method create_model (line 133) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 147) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _partition_inds (line 151) | def _partition_inds(n: int, num_parts: int): method _get_sampler_args_for_scene_split (line 156) | def _get_sampler_args_for_scene_split( method train_task_sampler_args (line 193) | def train_task_sampler_args( method valid_task_sampler_args (line 218) | def valid_task_sampler_args( method test_task_sampler_args (line 244) | def test_task_sampler_args( FILE: projects/tutorials/pointnav_habitat_rgb_ddppo.py class PointNavHabitatRGBPPOTutorialExperimentConfig (line 44) | class PointNavHabitatRGBPPOTutorialExperimentConfig(ExperimentConfig): method tag (line 139) | def tag(cls): method training_pipeline (line 143) | def training_pipeline(cls, **kwargs): method machine_params (line 176) | def machine_params(self, mode="train", **kwargs): method create_model (line 219) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 231) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 234) | def train_task_sampler_args( method valid_task_sampler_args (line 251) | def valid_task_sampler_args( method test_task_sampler_args (line 272) | def test_task_sampler_args( FILE: projects/tutorials/pointnav_ithor_rgb_ddppo.py class PointNaviThorRGBPPOExperimentConfig (line 39) | class PointNaviThorRGBPPOExperimentConfig(ExperimentConfig): method tag (line 108) | def tag(cls): method training_pipeline (line 112) | def training_pipeline(cls, **kwargs): method machine_params (line 145) | def machine_params(self, mode="train", **kwargs): method create_model (line 193) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 205) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _partition_inds (line 210) | def _partition_inds(n: int, num_parts: int): method _get_sampler_args_for_scene_split (line 215) | def _get_sampler_args_for_scene_split( method train_task_sampler_args (line 260) | def train_task_sampler_args( method valid_task_sampler_args (line 287) | def valid_task_sampler_args( method test_task_sampler_args (line 313) | def test_task_sampler_args( FILE: projects/tutorials/running_inference_tutorial.py class PointNavRoboThorRGBPPOVizExperimentConfig (line 119) | class PointNavRoboThorRGBPPOVizExperimentConfig(PointNavRoboThorRGBPPOEx... method get_viz (line 138) | def get_viz(self, mode): method machine_params (line 176) | def machine_params(self, mode="train", **kwargs): FILE: projects/tutorials/training_a_pointnav_model.py class PointNavRoboThorRGBPPOExperimentConfig (line 133) | class PointNavRoboThorRGBPPOExperimentConfig(ExperimentConfig): method tag (line 296) | def tag(cls): method training_pipeline (line 311) | def training_pipeline(cls, **kwargs): method machine_params (line 351) | def machine_params(self, mode="train", **kwargs): method create_model (line 410) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 431) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method _partition_inds (line 443) | def _partition_inds(n: int, num_parts: int): method _get_sampler_args_for_scene_split (line 448) | def _get_sampler_args_for_scene_split( method train_task_sampler_args (line 505) | def train_task_sampler_args( method valid_task_sampler_args (line 532) | def valid_task_sampler_args( method test_task_sampler_args (line 558) | def test_task_sampler_args( FILE: scripts/build_docs.py class StringColors (line 23) | class StringColors: function render_file (line 46) | def render_file( function build_docs_for_file (line 124) | def build_docs_for_file( function build_docs (line 142) | def build_docs( function project_readme_paths_to_nav_structure (line 197) | def project_readme_paths_to_nav_structure(project_readmes): function pruned_nav_entries (line 230) | def pruned_nav_entries(nav_entries): function main (line 254) | def main(): FILE: scripts/dcommand.py function get_argument_parser (line 10) | def get_argument_parser(): function get_args (line 47) | def get_args(): function wrap_double (line 56) | def wrap_double(text): function wrap_single (line 60) | def wrap_single(text): function wrap_single_nested (line 64) | def wrap_single_nested(text, quote=r"'\''"): FILE: scripts/dconfig.py function get_argument_parser (line 7) | def get_argument_parser(): function get_args (line 53) | def get_args(): function wrap_double (line 62) | def wrap_double(text): function wrap_single (line 66) | def wrap_single(text): function wrap_single_nested (line 70) | def wrap_single_nested(text, quote=r"'\''"): FILE: scripts/dkill.py function get_argument_parser (line 10) | def get_argument_parser(): function get_args (line 39) | def get_args(): FILE: scripts/dmain.py function get_argument_parser (line 23) | def get_argument_parser(): function get_args (line 67) | def get_args(): function get_raw_args (line 76) | def get_raw_args(): function wrap_single (line 109) | def wrap_single(text): function wrap_single_nested (line 113) | def wrap_single_nested(text): function wrap_double (line 121) | def wrap_double(text): function id_generator (line 125) | def id_generator(size=4, chars=string.ascii_uppercase + string.digits): FILE: scripts/literate.py function get_literate_output_path (line 13) | def get_literate_output_path(file: TextIO) -> Optional[str]: function source_to_markdown (line 42) | def source_to_markdown(dot_path: str, summarize: bool = False): function _strip_empty_lines (line 83) | def _strip_empty_lines(lines: Sequence[str]) -> List[str]: function literate_python_to_markdown (line 100) | def literate_python_to_markdown(path: str) -> bool: FILE: scripts/release.py function make_package (line 7) | def make_package(name, verbose=False): FILE: scripts/startx.py function pci_records (line 14) | def pci_records(): function generate_xorg_conf (line 28) | def generate_xorg_conf(devices): function startx (line 68) | def startx(display=0): FILE: tests/hierarchical_policies/test_minigrid_conditional.py class MiniGridCondTestExperimentConfig (line 30) | class MiniGridCondTestExperimentConfig(ExperimentConfig): method tag (line 32) | def tag(cls) -> str: method create_model (line 45) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 57) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method train_task_sampler_args (line 60) | def train_task_sampler_args( method valid_task_sampler_args (line 70) | def valid_task_sampler_args( method test_task_sampler_args (line 80) | def test_task_sampler_args( method _get_sampler_args (line 90) | def _get_sampler_args(self, process_ind: int, mode: str) -> Dict[str, ... method make_env (line 129) | def make_env(*args, **kwargs): method machine_params (line 133) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: method training_pipeline (line 140) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: class TestMiniGridCond (line 177) | class TestMiniGridCond: method test_train (line 178) | def test_train(self, tmpdir): FILE: tests/manipulathor_plugin/test_utils.py class TestArmCalculationUtils (line 6) | class TestArmCalculationUtils(object): method test_translation_functions (line 7) | def test_translation_functions(self): FILE: tests/mapping/test_ai2thor_mapping.py class TestAI2THORMapSensors (line 40) | class TestAI2THORMapSensors(object): method setup_path_for_use_with_rearrangement_project (line 41) | def setup_path_for_use_with_rearrangement_project(self) -> bool: method test_binned_and_semantic_mapping (line 66) | def test_binned_and_semantic_mapping(self, tmpdir): method test_pretrained_rearrange_walkthrough_mapping_agent (line 315) | def test_pretrained_rearrange_walkthrough_mapping_agent(self, tmpdir): FILE: tests/multiprocessing/test_frozen_attribs.py class MyConfig (line 12) | class MyConfig(ExperimentConfig): method tag (line 16) | def tag(cls) -> str: method training_pipeline (line 20) | def training_pipeline(cls, **kwargs) -> TrainingPipeline: method create_model (line 24) | def create_model(cls, **kwargs) -> nn.Module: method make_sampler_fn (line 28) | def make_sampler_fn(cls, **kwargs) -> TaskSampler: method my_var_is (line 31) | def my_var_is(self, val): class MySpecConfig (line 36) | class MySpecConfig(MyConfig): method machine_params (line 40) | def machine_params(cls, mode="train", **kwargs) -> Dict[str, Any]: method tag (line 44) | def tag(cls) -> str: class TestFrozenAttribs (line 51) | class TestFrozenAttribs(object): method test_frozen_inheritance (line 52) | def test_frozen_inheritance(self): method my_func (line 83) | def my_func(config, val): method test_frozen_experiment_config (line 86) | def test_frozen_experiment_config(self): FILE: tests/sync_algs_cpu/test_to_to_obj_trains.py class FixedConstantLoss (line 37) | class FixedConstantLoss(AbstractActorCriticLoss): method __init__ (line 38) | def __init__(self, name: str, value: float): method loss (line 43) | def loss( # type: ignore class SillyStorage (line 51) | class SillyStorage(ExperienceStorage, StreamingStorageMixin): method __init__ (line 52) | def __init__(self, values_to_return: List[float], repeats: List[int]): method initialize (line 58) | def initialize(self, *, observations: ObservationType, **kwargs): method add (line 61) | def add( method to (line 73) | def to(self, device: torch.device): method set_partition (line 76) | def set_partition(self, index: int, num_parts: int): method total_experiences (line 80) | def total_experiences(self) -> int: method total_experiences (line 84) | def total_experiences(self, value: int): method next_batch (line 87) | def next_batch(self) -> Dict[str, Any]: method reset_stream (line 99) | def reset_stream(self): method empty (line 102) | def empty(self) -> bool: class AverageBatchValueLoss (line 106) | class AverageBatchValueLoss(GenericAbstractLoss): method loss (line 107) | def loss( class PPOBabyAIGoToObjTestExperimentConfig (line 126) | class PPOBabyAIGoToObjTestExperimentConfig(PPOBabyAIGoToObjExperimentCon... method tag (line 130) | def tag(cls): method machine_params (line 134) | def machine_params(cls, mode="train", **kwargs): method training_pipeline (line 148) | def training_pipeline(cls, **kwargs): method valid_task_sampler_args (line 206) | def valid_task_sampler_args( class RedirectOutput (line 220) | class RedirectOutput: method __init__ (line 221) | def __init__(self, capsys: Optional, capfd: Optional): method get_output (line 232) | def get_output(self): method __enter__ (line 235) | def __enter__(self): method __exit__ (line 247) | def __exit__(self, *args): class TestGoToObjTrains (line 261) | class TestGoToObjTrains: method test_ppo_trains (line 262) | def test_ppo_trains(self, capfd, tmpdir): FILE: tests/utils/test_inference_agent.py class TestInferenceAgent (line 46) | class TestInferenceAgent(object): method test_inference_agent_from_minigrid_config (line 47) | def test_inference_agent_from_minigrid_config(self): FILE: tests/utils/test_spaces.py class TestSpaces (line 12) | class TestSpaces(object): method same (line 32) | def same(a, b, bidx=None): method test_conversion (line 50) | def test_conversion(self): method test_flatten (line 59) | def test_flatten(self): method test_batched (line 86) | def test_batched(self): method test_tolist (line 100) | def test_tolist(self): FILE: tests/vision/test_pillow_rescaling.py class TestPillowRescaling (line 14) | class TestPillowRescaling(object): method _load_thor_img (line 15) | def _load_thor_img(self) -> np.ndarray: method _get_img_hash (line 22) | def _get_img_hash(self, img: np.ndarray) -> str: method _random_rgb_image (line 26) | def _random_rgb_image(self, width: int, height: int, seed: int) -> np.... method _random_depthmap (line 35) | def _random_depthmap( method test_scaler_rgb_thor (line 44) | def test_scaler_rgb_thor(self): method test_scaler_rgb_random (line 68) | def test_scaler_rgb_random(self): method test_scaler_depth_thor (line 87) | def test_scaler_depth_thor(self): method test_scaler_depth_random (line 112) | def test_scaler_depth_random(self):