随机数种子
This commit is contained in:
@@ -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)
|
||||||
|
Reference in New Issue
Block a user