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