2020-1-capstone-design1 / PET_Project1

Go to a project
Toggle navigation Toggle navigation pinning
김성주
0af1fa34 1 parent 033f4743

fixed minor error

Inline Side-by-side
Showing 2 changed files with 2 additions and 2 deletions
...@@ -114,7 +114,7 @@ with tf.Session() as sess: ...@@ -114,7 +114,7 @@ with tf.Session() as sess:
114 loss_total, loss_xy, loss_wh, loss_conf, loss_class = AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter() 114 loss_total, loss_xy, loss_wh, loss_conf, loss_class = AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter()
115 115
116 ### train part 116 ### train part
117 - for i in trange(args.train_batch_num+1): 117 + for i in trange(args.train_batch_num):
118 _, __y_pred, __y_true, __loss, __global_step, __lr = sess.run( 118 _, __y_pred, __y_true, __loss, __global_step, __lr = sess.run(
119 [train_op, y_pred, y_true, loss, global_step, learning_rate], 119 [train_op, y_pred, y_true, loss, global_step, learning_rate],
120 feed_dict={is_training: True}) 120 feed_dict={is_training: True})
......
...@@ -1770,7 +1770,7 @@ ...@@ -1770,7 +1770,7 @@
1770 " loss_total, loss_xy, loss_wh, loss_conf, loss_class = AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter()\n", 1770 " loss_total, loss_xy, loss_wh, loss_conf, loss_class = AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter(), AverageMeter()\n",
1771 "\n", 1771 "\n",
1772 " ### train part\n", 1772 " ### train part\n",
1773 - " for i in trange(train_batch_num+1):\n", 1773 + " for i in trange(train_batch_num):\n",
1774 " _, __y_pred, __y_true, __loss, __global_step, __lr = sess.run(\n", 1774 " _, __y_pred, __y_true, __loss, __global_step, __lr = sess.run(\n",
1775 " [train_op, y_pred, y_true, loss, global_step, learning_rate],\n", 1775 " [train_op, y_pred, y_true, loss, global_step, learning_rate],\n",
1776 " feed_dict={is_training: True})\n", 1776 " feed_dict={is_training: True})\n",
......