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 23:03:46 +08:00
parent 6da74f8de1
commit 06baf5cce0
2 changed files with 23 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
serve.py
View File

@@ -150,7 +150,7 @@ lock = Lock()
config = Config()
def train_models():
def train_models(job):
print("train")
lock.acquire()
np.random.seed(config.random_seed)
@@ -192,15 +192,15 @@ class MyHandler(BaseHTTPRequestHandler):
self.wfile.write(bytes(json.dumps(msg), "utf-8"))
elif req.path == "/train":
try:
job = query.get('job')[0]
gpu_model = query.get('gpu_model')[0]
time = query.get('time')[0]
t = Thread(target=train_models, name='train_models', args=(job,))
t.start()
msg = {'code': 1, 'error': "container not exist"}
except Exception as e:
msg = {'code': 2, 'error': str(e)}
#try:
job = query.get('job')[0]
gpu_model = query.get('gpu_model')[0]
time = query.get('time')[0]
t = Thread(target=train_models, name='train_models', args=(job,))
t.start()
msg = {'code': 1, 'error': "container not exist"}
#except Exception as e:
msg = {'code': 2, 'error': str(e)}
self.send_response(200)
self.send_header('Content-type', 'application/json')
self.end_headers()