Skip to content
Snippets Groups Projects
Commit 09ffb1b2 authored by Shinya Shiroshita's avatar Shinya Shiroshita
Browse files

bugfix

parent 536a6d95
No related branches found
Tags submission-v0.1.17
No related merge requests found
......@@ -187,6 +187,14 @@ class MineRLSQILBaselineAgent(MineRLAgentBase):
self.env = env
def load_agent(self):
"""
boundaries = cached_reward_boundary(
cache_dir=os.environ.get('BOUNDARY_CACHE'),
env_id=MINERL_GYM_ENV,
n_groups=OPTION_N_GROUPS,
random_state=KMEANS_SEED)
reward_channel_scale = 1. / boundaries[-1]
"""
self.agent = get_agent(
n_actions=self.env.action_space.n, arch=ARCH, n_input_channels=self.env.observation_space.shape[0],
final_epsilon=FINAL_EPSILON,
......@@ -238,12 +246,6 @@ def main():
random_state=KMEANS_SEED,
subtask_reward_max=SUBTASK_REWARD_MAX,
only_vector_converter=True)
boundaries = cached_reward_boundary(
cache_dir=os.environ.get('BOUNDARY_CACHE'),
env_id=args.env,
n_groups=args.option_n_groups,
random_state=args.seed)
reward_channel_scale = 1. / boundaries[-1]
def wrapper(env):
return wrap_env(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment