-
Notifications
You must be signed in to change notification settings - Fork 2
/
submission_config.py
40 lines (30 loc) · 1.37 KB
/
submission_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from agents.torchbeast_agent import TorchBeastAgent
from agents.rulebased_agent import RuleBasedAgent
from agents.random_agent import RandomAgent
from envs.wrappers import addtimelimitwrapper_fn_rl
################################################
# Import your own agent code #
# Set Submision_Agent to your agent #
# Set NUM_PARALLEL_ENVIRONMENTS as needed #
# Set submission_env_make_fn to your wrappers #
# Test with local_evaluation.py #
################################################
class SubmissionConfig:
## Add your own agent class
# AGENT = RuleBasedAgent
AGENT = TorchBeastAgent
## Change the NUM_ENVIRONMENTS as you need
## for example reduce it if your GPU doesn't fit
## Increasing above 32 is not advisable for the Nethack Challenge 2021
NUM_ENVIRONMENTS = 8
## Add a function that creates your nethack env
## Mainly this is to add wrappers
## Add your wrappers to envs/wrappers.py and change the name here
## IMPORTANT: Don't "call" the function, only provide the name
MAKE_ENV_FN = addtimelimitwrapper_fn_rl
#MAKE_ENV_FN = addtimelimitwrapper_fn_custom
class TestEvaluationConfig:
# Change this to locally check a different number of rollouts
# The AIcrowd submission evaluator will not use this
# It is only for your local evaluation
NUM_EPISODES = 4096