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: