diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 579c8e2..fad21af 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -248,7 +248,7 @@ - + diff --git a/README.md b/README.md index be165d8..b1c6d85 100644 --- a/README.md +++ b/README.md @@ -2,10 +2,10 @@ ## Feed -/feed?job=1&model=2&time=3&utilGPU=4&utilCPU=5&pre=1&main=2&post=3 +/feed?job=1&seq=1&value=3 ## train /train ## predict -/predict?job=1&model=2&time=3&utilGPU=4&utilCPU=5 \ No newline at end of file +/predict?job=1&seq=2 \ No newline at end of file diff --git a/serve.py b/serve.py index 03f32f9..0ff875d 100644 --- a/serve.py +++ b/serve.py @@ -234,7 +234,8 @@ class MyHandler(BaseHTTPRequestHandler): elif req.path == "/train": try: - t = Thread(target=train_models, name='train_models', args=()) + job = query.get('job')[0] + t = Thread(target=train_models, name='train_models', args=(job,)) t.start() msg = {'code': 0, 'error': ""} except Exception as e: