From f8425bfa3108e64a8f41d0e9160ed3ef77ae2121 Mon Sep 17 00:00:00 2001 From: Dipam Chakraborty <dipam@aicrowd.com> Date: Wed, 9 Jun 2021 18:22:15 +0000 Subject: [PATCH] Update submission_config to TorchBeastAgent --- submission_config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/submission_config.py b/submission_config.py index ba35352..2eb1e58 100644 --- a/submission_config.py +++ b/submission_config.py @@ -14,8 +14,8 @@ from envs.wrappers import addtimelimitwrapper_fn class SubmissionConfig: ## Add your own agent class - AGENT = CustomAgent - # AGENT = TorchBeastAgent + # AGENT = CustomAgent + AGENT = TorchBeastAgent ## Change the NUM_ENVIRONMENTS as you need -- GitLab