diff --git a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/src/mains/res50.py b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/src/mains/res50.py
index ccb3e77d090ff4cb4cf79aa956364f088d412af1..d71fc42ace1d70e03d6eaf49403da5cd86aab504 100644
--- a/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/src/mains/res50.py
+++ b/TensorFlow/built-in/cv/image_classification/ResNet50_ID0058_for_TensorFlow/src/mains/res50.py
@@ -57,7 +57,6 @@ from trainers import gpu_base_trainer as tr
 # from configs import res50_config as cfg
 from hyper_param import hyper_param as hp
 from layers import layers as ly
-import torch_npu
 import argparse
 
 def main():