diff --git a/PyTorch/dev/cv/image_classification/SSD-MobileNet_ID1936_for_PyTorch/train.py b/PyTorch/dev/cv/image_classification/SSD-MobileNet_ID1936_for_PyTorch/train.py
index 5b78a89b3e8706b8fb45ddfda1d8743a70409687..bc8ca196dccddb804f7a42fe5deb4da1c68befc2 100644
--- a/PyTorch/dev/cv/image_classification/SSD-MobileNet_ID1936_for_PyTorch/train.py
+++ b/PyTorch/dev/cv/image_classification/SSD-MobileNet_ID1936_for_PyTorch/train.py
@@ -40,6 +40,7 @@ Created on Sat Jun  10 15:45:16 2019
 
 @author: viswanatha
 """
+import torch
 import time
 import torch.backends.cudnn as cudnn
 import torch.optim
@@ -58,6 +59,8 @@ try:
 except ImportError:
     amp = None
 import apex
+torch.npu.set_start_fuzz_compile_step(3)
+
 NPU_CALCULATE_DEVICE = 0
 if os.getenv('NPU_CALCULATE_DEVICE') and str.isdigit(os.getenv('NPU_CALCULATE_DEVICE')):
     NPU_CALCULATE_DEVICE = int(os.getenv('NPU_CALCULATE_DEVICE'))
@@ -86,6 +89,7 @@ def train(train_loader, model, criterion, optimizer, epoch, grad_clip, args):
 
     # Batches
     for i, (images, boxes, labels, _) in enumerate(train_loader):
+        torch.npu.global_step_inc()
         data_time.update(time.time() - start)
         start_time = time.time()