diff --git a/PyTorch/dev/cv/image_classification/VGG16_ID0467_for_PyTorch/train.py b/PyTorch/dev/cv/image_classification/VGG16_ID0467_for_PyTorch/train.py
index 4e9f258d821b649c2d8862a155e08857fad000fa..f9045e4110ffb13624da10ddba10cf447f5aad5e 100644
--- a/PyTorch/dev/cv/image_classification/VGG16_ID0467_for_PyTorch/train.py
+++ b/PyTorch/dev/cv/image_classification/VGG16_ID0467_for_PyTorch/train.py
@@ -62,7 +62,7 @@ def train_one_epoch(model, criterion, optimizer, data_loader, device, epoch, pri
     for image, target in metric_logger.log_every(data_loader, print_freq, header):
         start_time = time.time()
         #image, target = image.to(device), target.to(device)
-        image, target = image.to(device), target.to(torch.int).to(device)
+        image, target = image.to(device, non_blocking=True), target.to(torch.int).to(device, non_blocking=True)
         output = model(image)
         loss = criterion(output, target)