hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

seed???

Browse Source
This commit is contained in:
Trinkle23897 2020-03-20 21:51:09 +08:00
parent 75364cd986
commit d64d78d769
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/pytest.yml vendored
View File

@ -37,4 +37,4 @@ jobs:
- name: Test with pytest
run: |
pip install pytest pytest-cov
pytest --cov tianshou
pytest test --cov tianshou

2
test/test_ppo.py
View File

@ -61,7 +61,7 @@ class Critic(nn.Module):
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument('--task', type=str, default='CartPole-v0')
parser.add_argument('--seed', type=int, default=1626)
parser.add_argument('--seed', type=int, default=0)
parser.add_argument('--buffer-size', type=int, default=20000)
parser.add_argument('--lr', type=float, default=3e-3)
parser.add_argument('--gamma', type=float, default=0.99)