Added number of episodes in val split

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

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