Skip to content
Snippets Groups Projects
Commit f8425bfa authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

Update submission_config to TorchBeastAgent

parent b30420f1
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
submission_config.py
+ 2
− 2
View file @ f8425bfa
...@@ -14,8 +14,8 @@ from envs.wrappers import addtimelimitwrapper_fn ...@@ -14,8 +14,8 @@ from envs.wrappers import addtimelimitwrapper_fn
class SubmissionConfig: class SubmissionConfig:
## Add your own agent class ## Add your own agent class
AGENT = CustomAgent # AGENT = CustomAgent
# AGENT = TorchBeastAgent AGENT = TorchBeastAgent
## Change the NUM_ENVIRONMENTS as you need ## Change the NUM_ENVIRONMENTS as you need
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment