随机数种子

This commit is contained in:
2023-03-01 01:58:28 +08:00
parent 9dd1b82d3f
commit 8ab70b5aa0

View File

@@ -280,7 +280,7 @@ def main_dev(opt):
wm_encoder = WatermarkEncoder() wm_encoder = WatermarkEncoder()
wm_encoder.set_watermark('bytes', wm.encode('utf-8')) wm_encoder.set_watermark('bytes', wm.encode('utf-8'))
# x # 生成图片数
batch_size = task['number'] batch_size = task['number']
if not opt.from_file: if not opt.from_file:
@@ -303,8 +303,8 @@ def main_dev(opt):
# x # x
start_code = None start_code = None
if opt.fixed_code: if task['seed']: #if opt.fixed_code:
start_code = torch.randn([task['number'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device) start_code = torch.randn([task['seed'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device)
# 更新进度 # 更新进度
update_task_status(task, "running", 0.8) update_task_status(task, "running", 0.8)