Skip to content

Commit

Permalink
I hate tensorflow
Browse files Browse the repository at this point in the history
  • Loading branch information
YunYang1994 authored and YunYang1994 committed Dec 30, 2018
1 parent fe123de commit f0f13cc
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions quick_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@
example = iterator.get_next()
images, *y_true = example
model = yolov3.yolov3(num_classes)

with tf.variable_scope('yolov3'):
y_pred = model.forward(images, is_training=False)
loss = model.compute_loss(y_pred, y_true)
Expand Down

0 comments on commit f0f13cc

Please sign in to comment.