diff --git a/habitat_baselines/common/base_trainer.py b/habitat_baselines/common/base_trainer.py index 3af1209d6328d370c8510d8a8b6ae283a55561e1..407afabf47144b8e89e02b92745c82ab24d05341 100644 --- a/habitat_baselines/common/base_trainer.py +++ b/habitat_baselines/common/base_trainer.py @@ -123,6 +123,7 @@ class BaseRLTrainer(BaseTrainer): """ config = self.config.clone() + config.defrost() ckpt_cmd_opts = checkpoint_config.CMD_TRAILING_OPTS eval_cmd_opts = config.CMD_TRAILING_OPTS diff --git a/habitat_baselines/config/objectnav/ddppo_objectnav.yaml b/habitat_baselines/config/objectnav/ddppo_objectnav.yaml index c9d30f2245e0df02425af9cf873865f8390861e2..1e759ad89264bc52b4eda4560d80c49903decac1 100644 --- a/habitat_baselines/config/objectnav/ddppo_objectnav.yaml +++ b/habitat_baselines/config/objectnav/ddppo_objectnav.yaml @@ -6,10 +6,10 @@ TORCH_GPU_ID: 1 VIDEO_OPTION: ["disk", "tensorboard"] TENSORBOARD_DIR: "/checkpoint/maksymets/logs/habitat_baselines/ddppo/obj_nav_mp3d" VIDEO_DIR: "data/video_dir" -TEST_EPISODE_COUNT: 2 +TEST_EPISODE_COUNT: 2184 EVAL_CKPT_PATH_DIR: "data/new_checkpoints" NUM_PROCESSES: 2 -SENSORS: ["DEPTH_SENSOR"] +SENSORS: ["DEPTH_SENSOR", "RGB_SENSOR"] CHECKPOINT_FOLDER: "data/new_checkpoints" NUM_UPDATES: 10000 LOG_INTERVAL: 10