2020-1-capstone-design1 / PET_Project1

Go to a project
Toggle navigation Toggle navigation pinning
김성주
71958c29 1 parent 95f7f316

fixed batch loop

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