Merge pull request awjuliani#10 from wwxFromTju/master

Fix model loading behavior
GenesisXYL · Mar 2, 2017 · 1447dc5 · 1447dc5
2 parents fbeae7e + f711b86
commit 1447dc5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Double-Dueling-DQN.ipynb b/Double-Dueling-DQN.ipynb
@@ -278,11 +278,11 @@
     "    os.makedirs(path)\n",
     "\n",
     "with tf.Session() as sess:\n",
+    "    sess.run(init)\n",
     "    if load_model == True:\n",
     "        print 'Loading Model...'\n",
     "        ckpt = tf.train.get_checkpoint_state(path)\n",
     "        saver.restore(sess,ckpt.model_checkpoint_path)\n",
-    "    sess.run(init)\n",
     "    updateTarget(targetOps,sess) #Set the target network to be equal to the primary network.\n",
     "    for i in range(num_episodes):\n",
     "        episodeBuffer = experience_buffer()\n",
@@ -371,6 +371,7 @@
   }
  ],
  "metadata": {
+  "anaconda-cloud": {},
   "kernelspec": {
    "display_name": "Python 2",
    "language": "python",