Skip to content

Commit f8892e7

Browse files
committed
Dont' allow variable horizon for AIRL by default.
1 parent 75b9c6d commit f8892e7

File tree

4 files changed

+4
-1
lines changed

4 files changed

+4
-1
lines changed

src/imitation_cli/algorithm_configurations/airl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,4 @@ class Config:
3030
init_tensorboard: bool = False
3131
init_tensorboard_graph: bool = False
3232
debug_use_ground_truth: bool = False
33-
allow_variable_horizon: bool = True # TODO: true just for debugging
33+
allow_variable_horizon: bool = False

src/imitation_cli/config/airl_optuna.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ airl:
1616
demo_batch_size: 128
1717
demonstrations:
1818
total_timesteps: 10
19+
allow_variable_horizon: true
1920

2021
hydra:
2122
mode: MULTIRUN

src/imitation_cli/config/airl_run.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,3 +16,4 @@ airl:
1616
demo_batch_size: 128
1717
demonstrations:
1818
total_timesteps: 10
19+
allow_variable_horizon: true

src/imitation_cli/config/airl_sweep_env_and_rewardnet.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ airl:
1515
demo_batch_size: 128
1616
demonstrations:
1717
total_timesteps: 10
18+
allow_variable_horizon: true
1819

1920
hydra:
2021
mode: MULTIRUN

0 commit comments

Comments
 (0)