guru001 commited on
Commit
eb9d4eb
1 Parent(s): 2e6f614

model issue fixed

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -48,13 +48,13 @@ class api():
48
 
49
  # import meta from directory
50
  # import_meta = tf.train.import_meta_graph('{}.meta'.format(self.model_name))
51
- import_meta = tf.train.import_meta_graph(os.path.join('signs_api','{}.meta'.format(self.model_name)))
52
 
53
  with tf.Session() as sess:
54
 
55
  # tf.train.latest_checkpoint(<dir>) also works
56
 
57
- import_meta.restore(sess,'{}.ckpt'.format( os.path.join('signs_api',self.model_name) ) )
58
 
59
  # W1_val = sess.graph.get_tensor_by_name('W1:0')
60
 
 
48
 
49
  # import meta from directory
50
  # import_meta = tf.train.import_meta_graph('{}.meta'.format(self.model_name))
51
+ import_meta = tf.train.import_meta_graph(os.path.join('{}.meta'.format(self.model_name)))
52
 
53
  with tf.Session() as sess:
54
 
55
  # tf.train.latest_checkpoint(<dir>) also works
56
 
57
+ import_meta.restore(sess,'{}.ckpt'.format( os.path.join(self.model_name) ) )
58
 
59
  # W1_val = sess.graph.get_tensor_by_name('W1:0')
60