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

update

Browse Source
This commit is contained in:
Newnius
2020-05-02 01:38:39 +08:00
parent f3cef4f20a
commit e6bc92507c
2 changed files with 20 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
serve.py
View File

@@ -177,13 +177,12 @@ def train_models():
train_X, valid_X, train_Y, valid_Y = data_gainer.get_train_and_valid_data()
print(train_X, valid_X, train_Y, valid_Y)
print(train_X.shape[0])
if train_X.shape[0] < 500:
config.batch_size = 32
if train_X.shape[0] < 200:
config.batch_size = 16
print(train_X[:, :1], valid_X[:, :1], train_Y, valid_Y)
train(config, train_X[:, :1], train_Y, valid_X[:, :1], valid_Y)
lock.release()