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

add files

Browse Source
This commit is contained in:
Newnius
2020-04-29 22:54:37 +08:00
parent 5507bd66a8
commit 7c6309f010
2 changed files with 87 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
serve.py
View File

@@ -58,7 +58,7 @@ class Config:
if not os.path.exists(figure_save_path):
os.mkdir(figure_save_path)
used_frame = frame
used_frame = "tensorflow"
model_postfix = {"pytorch": ".pth", "keras": ".h5", "tensorflow": ".ckpt"}
model_name = "model_" + continue_flag + used_frame + model_postfix[used_frame]
@@ -148,7 +148,6 @@ def draw(config, origin_data, predict_norm_data):
PORT_NUMBER = 8000
lock = Lock()
config = Config()
frame = "tensorflow"
def train_models():