Spaces:
Sleeping
Sleeping
model issue fixed
Browse files
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('
|
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(
|
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 |
|