sizes error

Author: aamini

lab3/Lab3_rl.ipynb

@@ -686,7 +686,7 @@
         "def create_pong_model():\n",
         "  model = tf.keras.models.Sequential([\n",
         "      # Define and reshape inputs\n",
-        "      tf.keras.layers.InputLayer(input_shape=(80, 80, 1), dtype=tf.float32),\n",
+        "      tf.keras.layers.InputLayer(input_shape=(6400,), dtype=tf.float32),\n",
         "      tf.keras.layers.Reshape((80, 80, 1)),\n",
         "      \n",
         "      # Convolutional layers\n",
@@ -871,6 +871,7 @@
         "\n",
         "smoothed_reward = util.LossHistory(smoothing_factor=0.9)\n",
         "plotter = util.PeriodicPlotter(sec=5, xlabel='Iterations', ylabel='Rewards')\n",
+        "memory = Memory()\n",
         "\n",
         "for i_episode in range(MAX_ITERS):\n",
         "\n",