DEBUG:随机数

This commit is contained in:
2023-03-01 02:21:19 +08:00
parent 90f5f92abb
commit a8d597ea6d

View File

@@ -316,6 +316,7 @@ def main_dev(opt):
start_code = None start_code = None
if task['seed']: #if opt.fixed_code: 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) start_code = torch.randn([task['seed'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device)
print("start_code:", start_code)
# 更新进度 # 更新进度
update_task_status(task, "running", 0.8) update_task_status(task, "running", 0.8)