Skip to content

Commit

Permalink
small fix
Browse files Browse the repository at this point in the history
  • Loading branch information
ZiyueWang25 committed Oct 6, 2023
1 parent f8ebbc4 commit 33162b2
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion examples/train_dagger_atari_interactive_policy.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,6 @@ def lr_schedule(_: float):
features_extractor_class=torch_layers.FlattenExtractor,
)
wrapped_policy = obs_update_wrapper.RemoveHR(policy, lr_schedule=lr_schedule)
print(wrapped_policy.net_arch)

bc_trainer = bc.BC(
observation_space=env.observation_space,
Expand Down

0 comments on commit 33162b2

Please sign in to comment.