from typing import List, Optional, Union

from habitat.config.default import Config as CN
from habitat.config.default import get_config

_C = get_config()
_C.defrost()

# -----------------------------------------------------------------------------
# GPS SENSOR
# -----------------------------------------------------------------------------
_C.TASK.GLOBAL_GPS_SENSOR = CN()
_C.TASK.GLOBAL_GPS_SENSOR.TYPE = "GlobalGPSSensor"
_C.TASK.GLOBAL_GPS_SENSOR.DIMENSIONALITY = 3
# -----------------------------------------------------------------------------
# ORACLE ACTION SENSOR
# -----------------------------------------------------------------------------
_C.TASK.ORACLE_ACTION_SENSOR = CN()
_C.TASK.ORACLE_ACTION_SENSOR.TYPE = "OracleActionSensor"
_C.TASK.ORACLE_ACTION_SENSOR.GOAL_RADIUS = 0.5
# -----------------------------------------------------------------------------
# VLN ORACLE ACTION SENSOR
# -----------------------------------------------------------------------------
_C.TASK.VLN_ORACLE_ACTION_SENSOR = CN()
_C.TASK.VLN_ORACLE_ACTION_SENSOR.TYPE = "VLNOracleActionSensor"
_C.TASK.VLN_ORACLE_ACTION_SENSOR.GOAL_RADIUS = 0.5
# -----------------------------------------------------------------------------
# VLN ORACLE PROGRESS SENSOR
# -----------------------------------------------------------------------------
_C.TASK.VLN_ORACLE_PROGRESS_SENSOR = CN()
_C.TASK.VLN_ORACLE_PROGRESS_SENSOR.TYPE = "VLNOracleProgressSensor"
Example #2
0
from typing import List, Optional, Union

from habitat.config.default import (
    CONFIG_FILE_SEPARATOR,
)
from habitat.config.default import Config as CN
from habitat.config.default import get_config

_C = get_config()
_C.defrost()

# -----------------------------------------------------------------------------
# # SIMULATOR
# -----------------------------------------------------------------------------
_C.SIMULATOR.ACTION_SPACE_CONFIG = "pyrobotnoisy"
_C.SIMULATOR.NOISE_MODEL = CN()
_C.SIMULATOR.NOISE_MODEL.ROBOT = "LoCoBot"
_C.SIMULATOR.NOISE_MODEL.CONTROLLER = "ILQR"
_C.SIMULATOR.NOISE_MODEL.NOISE_MULTIPLIER = 1.0
# -----------------------------------------------------------------------------
# # TASK SENSORS
# -----------------------------------------------------------------------------
# -----------------------------------------------------------------------------
# STARTING COORDINATES SENSOR
# -----------------------------------------------------------------------------
_C.TASK.STARTING_COORDS_SENSOR = CN()
_C.TASK.STARTING_COORDS_SENSOR.TYPE = "StartingCoordinatesSensor"
# -----------------------------------------------------------------------------
# SCENE NAME SENSOR
# -----------------------------------------------------------------------------
_C.TASK.SCENE_NAME_SENSOR = CN()
from habitat.config.default import Config as CN
from habitat.config.default import get_config
from habitat.config.default import (
    CONFIG_FILE_SEPARATOR,
    DEFAULT_CONFIG_DIR,
)

_C = get_config()
_C.defrost()

# -----------------------------------------------------------------------------
# # SIMULATOR
# -----------------------------------------------------------------------------
_C.SIMULATOR.ACTION_SPACE_CONFIG = "pyrobotnoisy"
_C.SIMULATOR.NOISE_MODEL = CN()
_C.SIMULATOR.NOISE_MODEL.ROBOT = "LoCoBot"
_C.SIMULATOR.NOISE_MODEL.CONTROLLER = "ILQR"
_C.SIMULATOR.NOISE_MODEL.NOISE_MULTIPLIER = 1.0
# -----------------------------------------------------------------------------
# # TASK SENSORS
# -----------------------------------------------------------------------------
# -----------------------------------------------------------------------------
# GT POSE SENSOR
# -----------------------------------------------------------------------------
_C.TASK.GT_POSE_SENSOR = CN()
_C.TASK.GT_POSE_SENSOR.TYPE = "GTPoseSensor"
# -----------------------------------------------------------------------------
# NOISY POSE SENSOR
# -----------------------------------------------------------------------------
_C.TASK.NOISY_POSE_SENSOR = CN()
from typing import List, Optional, Union

from habitat.config.default import Config as CN

from habitat_extensions.config.default import get_extended_config as get_task_config

# -----------------------------------------------------------------------------
# EXPERIMENT CONFIG
# -----------------------------------------------------------------------------
_C = CN()
_C.BASE_TASK_CONFIG_PATH = "habitat_extensions/config/vlnce_task.yaml"
_C.TASK_CONFIG = CN()  # task_config will be stored as a config node
_C.CMD_TRAILING_OPTS = []  # store command line options as list of strings
_C.TRAINER_NAME = "dagger"
_C.ENV_NAME = "VLNCEDaggerEnv"
_C.SIMULATOR_GPU_ID = 0
_C.TORCH_GPU_ID = 0
_C.NUM_PROCESSES = 4
_C.VIDEO_OPTION = []  # options: "disk", "tensorboard"
_C.VIDEO_DIR = "videos/debug"
_C.TENSORBOARD_DIR = "data/tensorboard_dirs/debug"
_C.SENSORS = ["RGB_SENSOR", "DEPTH_SENSOR"]
_C.CHECKPOINT_FOLDER = "data/checkpoints"
_C.LOG_FILE = "train.log"
_C.EVAL_CKPT_PATH_DIR = "data/checkpoints"  # path to ckpt or path to ckpts dir

# -----------------------------------------------------------------------------
# EVAL CONFIG
# -----------------------------------------------------------------------------
_C.EVAL = CN()
# The split to evaluate on