ydin0771 commited on
Commit
004447e
β€’
1 Parent(s): a482138

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -41,7 +41,7 @@ def predict(image, question):
41
  with tf.Session(config=sessionConfig) as sess:
42
  sess.graph.finalize()
43
 
44
- epoch = loadWeights(sess, saver, init)
45
  emaSaver.restore(sess, config.weightsFile(epoch))
46
 
47
  evalRes = model.runBatch(sess, data['data'], data['image'], False)
 
41
  with tf.Session(config=sessionConfig) as sess:
42
  sess.graph.finalize()
43
 
44
+ epoch = 25
45
  emaSaver.restore(sess, config.weightsFile(epoch))
46
 
47
  evalRes = model.runBatch(sess, data['data'], data['image'], False)