Skip to content

Commit 2b7c267

Browse files
UbuntuUbuntu
authored andcommitted
This fixes NVlabs#6
1 parent 49dca59 commit 2b7c267

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

ga3c/Config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ class Config:
108108
# Epsilon (regularize policy lag in GA3C)
109109
LOG_EPSILON = 1e-6
110110
# Training min batch size - increasing the batch size increases the stability of the algorithm, but make learning slower
111-
TRAINING_MIN_BATCH_SIZE = 00
111+
TRAINING_MIN_BATCH_SIZE = 0
112112

113113
#########################################################################
114114
# Log and save

ga3c/ThreadTrainer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,9 @@ def run(self):
4747
if batch_size == 0:
4848
x__ = x_; r__ = r_; a__ = a_
4949
else:
50-
np.concatenate((x__, x_))
51-
np.concatenate((r__, r_))
52-
np.concatenate((a__, a_))
50+
x__ = np.concatenate((x__, x_))
51+
r__ = np.concatenate((r__, r_))
52+
a__ = np.concatenate((a__, a_))
5353
batch_size += x_.shape[0]
5454

5555
if Config.TRAIN_MODELS:

0 commit comments

Comments
 (0)