From 812df9245bc08d449b7cb61403978e9cfe85abb3 Mon Sep 17 00:00:00 2001
From: Oleksandr Maksymets <maksymets@gmail.com>
Date: Fri, 7 Feb 2020 10:46:26 -0800
Subject: [PATCH] Added number of episodes in val split
---
habitat_baselines/common/base_trainer.py | 1 +
habitat_baselines/config/objectnav/ddppo_objectnav.yaml | 4 ++--
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/habitat_baselines/common/base_trainer.py b/habitat_baselines/common/base_trainer.py
index 3af1209d6..407afabf4 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 c9d30f224..1e759ad89 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
--
GitLab