diff --git a/server.py b/server.py index 8b082a9..3bba55d 100644 --- a/server.py +++ b/server.py @@ -253,6 +253,7 @@ def main_dev(opt): while True: for task in get_tasks(): # 遍历 tasks 返回 dict print('task:', task) # 打印任务 + seed_everything(task['seed']) # 设置随机种子 # 如果模型不同,重新加载模型(注意释放内存) if task['ckpt'] != model_name: @@ -314,8 +315,8 @@ def main_dev(opt): # x start_code = None - if task['seed']: #if opt.fixed_code: - start_code = torch.randn([task['seed'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device) + if opt.fixed_code: + start_code = torch.randn([task['number'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device) print("start_code:", start_code) # 更新进度