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-29 18:39:51 +08:00
parent a8a1a06d75
commit 34f2824db8
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
model/model_tensorflow.py
View File

@@ -46,7 +46,7 @@ def train(config, train_X, train_Y, valid_X, valid_Y):
bad_epoch = 0
for epoch in range(config.epoch):
print("Epoch {}/{}".format(epoch, config.epoch))
# 训练
train_loss_array = []
for step in range(train_len // config.batch_size):
feed_dict = {model.X: train_X[step * config.batch_size: (step + 1) * config.batch_size],
@@ -54,7 +54,7 @@ def train(config, train_X, train_Y, valid_X, valid_Y):
train_loss, _ = sess.run([model.loss, model.optim], feed_dict=feed_dict)
train_loss_array.append(train_loss)
# 验证与早停
valid_loss_array = []
for step in range(valid_len // config.batch_size):
feed_dict = {model.X: valid_X[step * config.batch_size: (step + 1) * config.batch_size],