From 544a0f7b6e2b5a7e8f70b0ca8e02e45f8fd706a2 Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 18:53:16 +0800 Subject: [PATCH] update --- .idea/workspace.xml | 2 +- serve.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 50b56be..d6e4d2f 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -230,7 +230,7 @@ - + diff --git a/serve.py b/serve.py index eb5e30c..bb1085a 100644 --- a/serve.py +++ b/serve.py @@ -126,7 +126,7 @@ def train_models(job): train_trimmed = train_scaled[t1:, :] model = fit_lstm(train_trimmed, batch_size, 30, 4) - model.saver.save('./data/checkpoint-' + job) + model.save('./data/checkpoint-' + job) models[job]['scaler'] = scaler models[job]['batch_size'] = batch_size