Commit f8425bfa authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

Update submission_config to TorchBeastAgent

parent b30420f1
......@@ -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
......
Markdown is supported
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