added benchmark task Crafter

2023-06-18 00:02:22 +09:00
parent 9c58ab62c0
commit 5dce8cf13b
6 changed files with 111 additions and 5 deletions
--- a/dreamer.py
+++ b/dreamer.py
@@ -211,6 +211,11 @@ def make_env(config, logger, mode, train_eps, eval_eps):
            task, mode if "train" in mode else "test", config.action_repeat
        )
        env = wrappers.OneHotAction(env)
+    elif suite == "crafter":
+        import envs.crafter as crafter
+
+        env = crafter.Crafter(task, config.size)
+        env = wrappers.OneHotAction(env)
    else:
        raise NotImplementedError(suite)
    env = wrappers.TimeLimit(env, config.time_limit)