YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-15 17:06:44 +00:00
Code Issues Releases Wiki Activity

add files

Browse Source
This commit is contained in:
Newnius
2020-04-29 18:54:53 +08:00
parent 30ea099b2b
commit c01add193e
2 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
model/model_tensorflow.py
View File

@@ -86,7 +86,7 @@ def predict(config, test_X):
test_len = len(test_X)
with tf.Session() as sess:
module_file = tf.train.latest_checkpoint(config.model_save_path + config.model_name)
module_file = tf.train.latest_checkpoint(config.model_save_path)
model.saver.restore(sess, module_file)
result = np.zeros((test_len * config.time_step, config.output_size))