随机数种子

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: