From fd861082073cc538b2dc867251806b145c60e7ab Mon Sep 17 00:00:00 2001
From: Dipam Chakraborty <dipam@aicrowd.com>
Date: Wed, 9 Jun 2021 18:31:08 +0000
Subject: [PATCH] Revert "Update submission_config to TorchBeastAgent"
This reverts commit f8425bfa3108e64a8f41d0e9160ed3ef77ae2121
---
submission_config.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/submission_config.py b/submission_config.py
index 2eb1e58..ba35352 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