From 7dde306d382b51f195ca5a655ed707182674037c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Wed, 1 Mar 2023 03:01:05 +0800 Subject: [PATCH] =?UTF-8?q?=E9=9A=8F=E6=9C=BA=E6=95=B0=E7=A7=8D=E5=AD=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/server.py b/server.py index 3bba55d..2e047d3 100644 --- a/server.py +++ b/server.py @@ -242,6 +242,7 @@ def get_tasks(tasks:list=[]): time.sleep(2) return tasks +import random def main_dev(opt): signal.signal(signal.SIGINT, quit) @@ -253,7 +254,9 @@ def main_dev(opt): while True: for task in get_tasks(): # 遍历 tasks 返回 dict print('task:', task) # 打印任务 - seed_everything(task['seed']) # 设置随机种子 + + # 设置随机种子(如果为空, 则使用随机数) + seed_everything(task['seed'] if task['seed'] else random.randint(0, 100000)) # 如果模型不同,重新加载模型(注意释放内存) if task['ckpt'] != model_name: