Amend
This commit is contained in:
@@ -340,7 +340,7 @@ class Train(Train_Base):
|
||||
def train(self, args):
|
||||
|
||||
# --------------------------------------- Learning parameters
|
||||
n_envs = min(12, os.cpu_count())
|
||||
n_envs = min(8, os.cpu_count())
|
||||
n_steps_per_env = 1024 # RolloutBuffer is of size (n_steps_per_env * n_envs)
|
||||
minibatch_size = 64 # should be a factor of (n_steps_per_env * n_envs)
|
||||
total_steps = 50000000
|
||||
|
||||
Reference in New Issue
Block a user