Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train experts for AIRL environments #68

Merged
merged 15 commits into from
Aug 15, 2019
+133 −52
Merged

Train experts for AIRL environments #68

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
55ffa61
scripts.config: Organize named configs
shwang Jul 31, 2019
b6ff4a5
scripts.config: Add custom env named configs
shwang Jul 31, 2019
e22a089
mujoco_experts.sh: Train custom env experts
shwang Jul 31, 2019
366d4e6
Rename mujoco_experts.sh => train_experts.sh
shwang Jul 31, 2019
ca6ca84
Fix lint/Travis/imports
shwang Jul 31, 2019
04390c5
train_experts: save to train_experts/
shwang Aug 5, 2019
bf759a9
setup.py: Install stable_baselines from master
shwang Aug 5, 2019
7eceb5f
Accommodate PPO1 import fail
shwang Aug 13, 2019
e156841
Merge branch 'master' into airl_data_collect
shwang Aug 13, 2019
f979c35
address comments
shwang Aug 14, 2019
c16bca7
Address comments
shwang Aug 15, 2019
0d2a0cb
Fix configs
shwang Aug 15, 2019
9a16e3f
lint
shwang Aug 15, 2019
f981990
Merge branch 'master' into airl_data_collect
AdamGleave Aug 15, 2019
dd6ae1c
Fix bad merge: add back in fast
AdamGleave Aug 15, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix lint/Travis/imports
  • Loading branch information
@shwang
shwang committed Jul 31, 2019
commit ca6ca843fe55f0762026bb3b7a54820b4c84c692
2 changes: 1 addition & 1 deletion src/imitation/scripts/config/data_collect.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import sacred

import imitation.examples.airl_envs
import imitation.examples.airl_envs # noqa: F401
from imitation.scripts.config.common import DEFAULT_BLANK_POLICY_KWARGS
from imitation.util import util

Expand Down
3 changes: 2 additions & 1 deletion src/imitation/scripts/config/train.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from stable_baselines.common import policies

from imitation import util
import imitation.examples.airl_envs # noqa: F401
from imitation.policies import base
from imitation.scripts.config.common import DEFAULT_BLANK_POLICY_KWARGS

Expand Down Expand Up @@ -149,7 +150,7 @@ def walker():

@train_ex.named_config
def two_d_maze():
env_name = "imitation/TwoDMaze-v0
env_name = "imitation/TwoDMaze-v0"


@train_ex.named_config
Expand Down
1 change: 0 additions & 1 deletion src/imitation/scripts/train.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
import tensorflow as tf
import tqdm

import imitation.examples.airl_envs
from imitation.scripts.config.train import train_ex
import imitation.util as util
from imitation.util.trainer import init_trainer
Expand Down