dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Testing SDE...

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-08-14 18:42:45 +02:00
parent bb1f9ecf2b
commit 639aae7f42
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test.py
View File

@ -20,11 +20,11 @@ root_path = '.'
def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=1_000_000, showRes=True, saveModel=True, n_eval_episodes=0):
env = gym.make(env_name)
use_sde = False
use_sde = True
ppo = PPO(
MlpPolicyPPO,
env,
projection=FrobeniusProjectionLayer(),
projection=BaseProjectionLayer(),
policy_kwargs={'dist_kwargs': {'neural_strength': Strength.FULL, 'cov_strength': Strength.FULL, 'parameterization_type':
ParametrizationType.CHOL, 'enforce_positive_type': EnforcePositiveType.ABS, 'prob_squashing_type': ProbSquashingType.NONE}},
verbose=0,
@ -149,5 +149,5 @@ if __name__ == '__main__':
# main('ColumbusJustState-v0')
# main('ColumbusStateWithBarriers-v0')
# full('ColumbusEasierObstacles-v0')
full('ColumbusStateWithBarriers-v0')
main('ColumbusSingle-v0')
# full('LunarLanderContinuous-v2')