app

app.py

@@ -51,10 +51,10 @@ device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cp
 print("=> using device: {}".format(device))
 config.device = device
 # torch.backends.cudnn.benchmark = True
-print("=> creating diffusion model")
+
 diffusion = DenoisingDiffusion(config, test=True)
 model = DiffusiveRestoration(diffusion, config)
-
+print("=> creating diffusion model")
 
 # 加载数据
 # seed = 61