gameui/drawing
2
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

随机数种子

Browse Source
This commit is contained in:
桜華
2023-03-01 03:01:05 +08:00
parent fa1a244944
commit 7dde306d38
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@@ -242,6 +242,7 @@ def get_tasks(tasks:list=[]):
time.sleep(2) time.sleep(2)
return tasks return tasks
import random
def main_dev(opt): def main_dev(opt):
signal.signal(signal.SIGINT, quit) signal.signal(signal.SIGINT, quit)
@@ -253,7 +254,9 @@ def main_dev(opt):
while True: while True:
for task in get_tasks(): # 遍历 tasks 返回 dict for task in get_tasks(): # 遍历 tasks 返回 dict
print('task:', task) # 打印任务 print('task:', task) # 打印任务
seed_everything(task['seed']) # 设置随机种子
# 设置随机种子(如果为空, 则使用随机数)
seed_everything(task['seed'] if task['seed'] else random.randint(0, 100000))
# 如果模型不同，重新加载模型(注意释放内存) # 如果模型不同，重新加载模型(注意释放内存)
if task['ckpt'] != model_name: if task['ckpt'] != model_name: