YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-06-06 22:51:55 +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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.idea/workspace.xml
View File

@ -134,7 +134,7 @@
<component name="PropertiesComponent">
<property name="WebServerToolWindowFactoryState" value="false" />
<property name="aspect.path.notification.shown" value="true" />
<property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588156719300" />
<property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588156758264" />
<property name="go.gopath.indexing.explicitly.defined" value="true" />
<property name="nodejs_interpreter_path.stuck_in_default_project" value="undefined stuck path" />
<property name="nodejs_npm_path_reset_for_default_project" value="true" />

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],