diff --git a/sim/a3c.py b/sim/a3c.py index a845be110..2646d15ad 100644 --- a/sim/a3c.py +++ b/sim/a3c.py @@ -67,7 +67,7 @@ def create_actor_network(self): split_2 = tflearn.conv_1d(inputs[:, 2:3, :], 128, 4, activation='relu') split_3 = tflearn.conv_1d(inputs[:, 3:4, :], 128, 4, activation='relu') split_4 = tflearn.conv_1d(inputs[:, 4:5, :A_DIM], 128, 4, activation='relu') - split_5 = tflearn.fully_connected(inputs[:, 4:5, -1], 128, activation='relu') + split_5 = tflearn.fully_connected(inputs[:, 5:6, -1], 128, activation='relu') split_2_flat = tflearn.flatten(split_2) split_3_flat = tflearn.flatten(split_3)