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

add files

Browse Source
This commit is contained in:
Newnius
2020-04-30 00:41:27 +08:00
parent a332ca27f0
commit b9418c7508
2 changed files with 27 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
model_tensorflow.py
View File

@@ -62,6 +62,8 @@ def train(config, train_X, train_Y, valid_X, valid_Y):
valid_loss = sess.run(model.loss, feed_dict=feed_dict)
valid_loss_array.append(valid_loss)
print(2)
print(valid_loss_array)
valid_loss_cur = np.mean(valid_loss_array)
print("The train loss is {:.4f}. ".format(np.mean(train_loss_array)),
"The valid loss is {:.4f}.".format(valid_loss_cur))