Fix windows env setup bugs and other typo. (#11)

This commit is contained in:
ShenDezhou 2020-03-31 17:22:32 +08:00 committed by GitHub
parent 98feb79057
commit 4da857d86e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 5 additions and 5 deletions

View File

@ -16,7 +16,7 @@ setup(
name='tianshou',
version=version,
description='A Library for Deep Reinforcement Learning',
long_description=open('README.md').read(),
long_description=open('README.md', encoding='utf8').read(),
long_description_content_type='text/markdown',
url='https://github.com/thu-ml/tianshou',
author='TSAIL',

View File

@ -74,7 +74,7 @@ def test_a2c(args=get_args()):
policy, train_envs, ReplayBuffer(args.buffer_size))
test_collector = Collector(policy, test_envs)
# log
writer = SummaryWriter(args.logdir + '/' + 'ppo')
writer = SummaryWriter(args.logdir + '/' + 'a2c')
def stop_fn(x):
return x >= env.spec.reward_threshold

View File

@ -73,7 +73,7 @@ def test_dqn(args=get_args()):
# policy.set_eps(1)
train_collector.collect(n_step=args.batch_size)
# log
writer = SummaryWriter(args.logdir + '/' + 'ppo')
writer = SummaryWriter(args.logdir + '/' + 'dqn')
def stop_fn(x):
return x >= env.spec.reward_threshold

View File

@ -122,7 +122,7 @@ def test_pg(args=get_args()):
policy, train_envs, ReplayBuffer(args.buffer_size))
test_collector = Collector(policy, test_envs)
# log
writer = SummaryWriter(args.logdir + '/' + 'ppo')
writer = SummaryWriter(args.logdir + '/' + 'pg')
def stop_fn(x):
return x >= env.spec.reward_threshold

View File

@ -1,7 +1,7 @@
from tianshou import data, env, utils, policy, trainer, \
exploration
__version__ = '0.2.0post2'
__version__ = '0.2.0.post2'
__all__ = [
'env',
'data',