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 02:53:47 +08:00
parent a8d597ea6d
commit fa1a244944
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@@ -253,6 +253,7 @@ 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']) # 设置随机种子
# 如果模型不同，重新加载模型(注意释放内存) # 如果模型不同，重新加载模型(注意释放内存)
if task['ckpt'] != model_name: if task['ckpt'] != model_name:
@@ -314,8 +315,8 @@ def main_dev(opt):
# x # x
start_code = None start_code = None
if task['seed']: #if opt.fixed_code: if opt.fixed_code:
start_code = torch.randn([task['seed'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device) start_code = torch.randn([task['number'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device)
print("start_code:", start_code) print("start_code:", start_code)
# 更新进度 # 更新进度