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 797263fda68829f489aa9441075f907671fdac2c..feefa5d01e2bc4542d359855640f8d32adc6a550 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)