Use h_size to determine final encoder layer

GenesisXYL · Mar 9, 2017 · e81f8d7 · e81f8d7
1 parent baff0a8
commit e81f8d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Double-Dueling-DQN.ipynb b/Double-Dueling-DQN.ipynb
@@ -128,7 +128,7 @@
     "        self.conv3 = slim.conv2d( \\\n",
     "            inputs=self.conv2,num_outputs=64,kernel_size=[3,3],stride=[1,1],padding='VALID', biases_initializer=None)\n",
     "        self.conv4 = slim.conv2d( \\\n",
-    "            inputs=self.conv3,num_outputs=512,kernel_size=[7,7],stride=[1,1],padding='VALID', biases_initializer=None)\n",
+    "            inputs=self.conv3,num_outputs=h_size,kernel_size=[7,7],stride=[1,1],padding='VALID', biases_initializer=None)\n",
     "        \n",
     "        #We take the output from the final convolutional layer and split it into separate advantage and value streams.\n",
     "        self.streamAC,self.streamVC = tf.split(self.conv4,2,3)\n",