|
@@ -308,12 +308,9 @@
|
|
|
" sess.run(optimizer, feed_dict={x: batch_x, y: batch_y,\n",
|
|
|
" seqlen: batch_seqlen})\n",
|
|
|
" if step % display_step == 0 or step == 1:\n",
|
|
|
- " # Calculate batch accuracy\n",
|
|
|
- " acc = sess.run(accuracy, feed_dict={x: batch_x, y: batch_y,\n",
|
|
|
+ " # Calculate batch accuracy & loss\n",
|
|
|
+ " acc, loss = sess.run([accuracy, cost], feed_dict={x: batch_x, y: batch_y,\n",
|
|
|
" seqlen: batch_seqlen})\n",
|
|
|
- " # Calculate batch loss\n",
|
|
|
- " loss = sess.run(cost, feed_dict={x: batch_x, y: batch_y,\n",
|
|
|
- " seqlen: batch_seqlen})\n",
|
|
|
" print(\"Step \" + str(step) + \", Minibatch Loss= \" + \\\n",
|
|
|
" \"{:.6f}\".format(loss) + \", Training Accuracy= \" + \\\n",
|
|
|
" \"{:.5f}\".format(acc))\n",
|