diff --git a/defaults/template-system.env b/defaults/template-system.env index 680f888a..8d00c64f 100644 --- a/defaults/template-system.env +++ b/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= DR_MINIO_IMAGE=latest diff --git a/docker/docker-compose-training.yml b/docker/docker-compose-training.yml index 0848f465..f6a31261 100644 --- a/docker/docker-compose-training.yml +++ b/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" diff --git a/scripts/training/prepare-config.py b/scripts/training/prepare-config.py index 3f692c6d..f2fe04c5 100755 --- a/scripts/training/prepare-config.py +++ b/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':