Add params for KVS overlay and sagemaker metrics (#179)

Co-authored-by: Lars Lorentz Ludvigsen <[email protected]>

defaults/template-system.env

@@ -10,6 +10,7 @@ DR_KINESIS_STREAM_NAME=
 DR_CAMERA_MAIN_ENABLE=True
 DR_CAMERA_SUB_ENABLE=False
 DR_CAMERA_KVS_ENABLE=True
+DR_ENABLE_EXTRA_KVS_OVERLAY=False
 DR_SIMAPP_SOURCE=awsdeepracercommunity/deepracer-simapp
 DR_SIMAPP_VERSION=<SIMAPP_VERSION_TAG>
 DR_MINIO_IMAGE=latest

docker/docker-compose-training.yml

@@ -25,7 +25,8 @@ services:
       - CUDA_VISIBLE_DEVICES=${DR_SAGEMAKER_CUDA_DEVICES:-}
       - MAX_MEMORY_STEPS=${DR_TRAIN_MAX_STEPS_PER_ITERATION:-}
       - TELEGRAF_HOST=${DR_TELEGRAF_HOST:-}
-      - TELEGRAF_PORT=${DR_TELEGRAF_PORT:-}      
+      - TELEGRAF_PORT=${DR_TELEGRAF_PORT:-}
+
     volumes:
       - "/var/run/docker.sock:/var/run/docker.sock"
       - "/tmp/sagemaker:/tmp/sagemaker"

scripts/training/prepare-config.py

@@ -52,6 +52,7 @@
 config['CAMERA_MAIN_ENABLE'] = os.environ.get('DR_CAMERA_MAIN_ENABLE', 'True')
 config['CAMERA_SUB_ENABLE'] = os.environ.get('DR_CAMERA_SUB_ENABLE', 'True')
 config['BEST_MODEL_METRIC'] = os.environ.get('DR_TRAIN_BEST_MODEL_METRIC', 'progress')
+config['ENABLE_EXTRA_KVS_OVERLAY'] = os.environ.get('DR_ENABLE_EXTRA_KVS_OVERLAY', 'False')
 
 # Object Avoidance
 if config['RACE_TYPE'] == 'OBJECT_AVOIDANCE':
@@ -158,6 +159,8 @@
             config.update({'REVERSE_DIR': os.environ.get('DR_TRAIN_REVERSE_DIRECTION', False)})
             config.update({'CAMERA_MAIN_ENABLE': os.environ.get('DR_CAMERA_MAIN_ENABLE', 'True')})
             config.update({'CAMERA_SUB_ENABLE': os.environ.get('DR_CAMERA_SUB_ENABLE', 'True')})  
+            config.update({'ENABLE_EXTRA_KVS_OVERLAY': os.environ.get('DR_ENABLE_EXTRA_KVS_OVERLAY', 'False')})
+
 
             # Update Object Avoidance parameters
             if config['RACE_TYPE'] == 'OBJECT_AVOIDANCE':