gomoku / LightZero /zoo /bsuite /envs /check_bsuite_config.py
zjowowen's picture
init space
079c32c
raw
history blame
955 Bytes
import bsuite
from bsuite import sweep
# List the configurations for the given experiment
for bsuite_id in sweep.BANDIT_NOISE:
env = bsuite.load_from_id(bsuite_id)
print('bsuite_id={}, settings={}, num_episodes={}'
.format(bsuite_id, sweep.SETTINGS[bsuite_id], env.bsuite_num_episodes))
for bsuite_id in sweep.CARTPOLE_SWINGUP:
env = bsuite.load_from_id(bsuite_id)
print('bsuite_id={}, settings={}, num_episodes={}'
.format(bsuite_id, sweep.SETTINGS[bsuite_id], env.bsuite_num_episodes))
for bsuite_id in sweep.MEMORY_LEN:
env = bsuite.load_from_id(bsuite_id)
print('bsuite_id={}, settings={}, num_episodes={}'
.format(bsuite_id, sweep.SETTINGS[bsuite_id], env.bsuite_num_episodes))
for bsuite_id in sweep.MEMORY_SIZE:
env = bsuite.load_from_id(bsuite_id)
print('bsuite_id={}, settings={}, num_episodes={}'
.format(bsuite_id, sweep.SETTINGS[bsuite_id], env.bsuite_num_episodes))