fixed experiment A script

danielbinschmid · Jan 26, 2024 · 542e92b · 542e92b
1 parent 1438791
commit 542e92b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agents/run_experiment_A.py b/agents/run_experiment_A.py
@@ -106,7 +106,7 @@ def run(output_folder=DEFAULT_OUTPUT_FOLDER,
     parser.add_argument('--action_type',          default=DEFAULT_TIMESTEPS,     type=str,           help='Either "one_d_rpm", "rpm" or "attitude"', metavar='')
     parser.add_argument('--train',          default=DEFAULT_TRAIN,     type=str2bool,           help='Whether to train (default: True)', metavar='')
     parser.add_argument('--test',          default=DEFAULT_TEST,     type=str2bool,           help='Whether to test (default: True)', metavar='')
-    parser.add_argument('--mode',          default=DEFAULT_MODE,     type=str2bool,           help='Experiment mode (default "UP")', metavar='')
+    parser.add_argument('--mode',          default=DEFAULT_MODE,     type=str,           help='Experiment mode (default "UP")', metavar='')
     ARGS = parser.parse_args()
 
     run(**vars(ARGS))
diff --git a/runnables/run_experiment_A.sh b/runnables/run_experiment_A.sh
@@ -9,7 +9,7 @@ gui="False"
 test="False"
 train="True"
 base_output_folder="exp-A"
-total_runs=10
+total_runs=1
 
 declare -a action_types=("attitude" "rpm")
 declare -a modes=("UP" "DOWN" "SIDEWAYS" "DIAGONAL_UP" "DIAGONAL_DOWN")