update PolicyGradient

This commit is contained in:
JohnJim0816
2020-11-23 12:13:34 +08:00
parent 4f192c7aac
commit 4cc12bf97f
5 changed files with 185 additions and 0 deletions

View File

@@ -0,0 +1,19 @@
#!/usr/bin/env python
# coding=utf-8
'''
Author: John
Email: johnjim0816@gmail.com
Date: 2020-11-22 23:25:37
LastEditor: John
LastEditTime: 2020-11-22 23:32:44
Discription: 存储参数
Environment:
'''
import argparse
def get_args():
'''训练参数'''
parser = argparse.ArgumentParser()
parser.add_argument("--train_eps", default=1200, type=int) # 训练的最大episode数目
parser.add_argument("--policy_lr", default=0.01, type=float) # 学习率
config = parser.parse_args()
return config