From 729e017affc01bf8192c47c00be8a8070d615186 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=89=E5=AE=8F=E6=A2=85?= <591861959@qq.com>
Date: Thu, 7 Apr 2022 08:23:19 +0000
Subject: [PATCH] update

---
 .../cv/semantic_segmentation/AttU_Net_for_PyTorch/solver.py   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/PyTorch/built-in/cv/semantic_segmentation/AttU_Net_for_PyTorch/solver.py b/PyTorch/built-in/cv/semantic_segmentation/AttU_Net_for_PyTorch/solver.py
index 797263fda6..feefa5d01e 100644
--- a/PyTorch/built-in/cv/semantic_segmentation/AttU_Net_for_PyTorch/solver.py
+++ b/PyTorch/built-in/cv/semantic_segmentation/AttU_Net_for_PyTorch/solver.py
@@ -191,8 +191,8 @@ class Solver(object):
                 # GT : Ground Truth
                 if i == 5:
                     start_time = time.time()
-                images = images.to(self.device)
-                GT = GT.to(self.device)
+                images = images.to(self.device,non_blocking=True)
+                GT = GT.to(self.device,non_blocking=True)
 
                 # SR : Segmentation Result
                 SR = self.unet(images)
-- 
Gitee