Skip to content

Commit

Permalink
Update hps
Browse files Browse the repository at this point in the history
  • Loading branch information
cswinter committed Oct 9, 2019
1 parent d5ca778 commit 21011a1
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions hyper_params.py
Expand Up @@ -5,15 +5,15 @@
class HyperParams:
def __init__(self):
# Optimizer
self.optimizer = 'RMSProp' # Optimizer
self.optimizer = 'RMSProp' # Optimizer
self.lr = 0.0001 # Learning rate
self.momentum = 0.9 # Momentum
self.weight_decay = 0.0001
self.bs = 512 # Batch size during optimization
self.bs = 2048 # Batch size during optimization
self.shuffle = True # Shuffle samples collected during rollout before optimization
self.vf_coef = 0.5 # Weighting of value function loss in optimization objective
self.max_grad_norm = 1.0 # Maximum gradient norm for gradient clipping
self.sample_reuse = 1 # Number of optimizer passes over samples collected during rollout
self.sample_reuse = 2 # Number of optimizer passes over samples collected during rollout

# Policy
self.depth = 4 # Number of hidden layers
Expand Down

0 comments on commit 21011a1

Please sign in to comment.