diff --git a/constants.py b/constants.py index 753328f..fb3db45 100644 --- a/constants.py +++ b/constants.py @@ -45,7 +45,12 @@ ######################################################################################################################## # Task specific hyperparameters +# Defaults +RENDER_DEPTH_IMAGE = False +RENDER_CLASS_IMAGE = False +RENDER_OBJECT_IMAGE = False +# Per task if TASK == 'navigation': RL = False SUPERVISED = True diff --git a/requirements.txt b/requirements.txt index 5822ea7..bce5c18 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,4 +14,4 @@ mss progressbar2 tensorflow-gpu==1.5.0 requests -ai2thor==0.0.32 +ai2thor==0.0.34