diff --git a/noise_cifar_train.py b/noise_cifar_train.py
index fcff6269b01a7b306405215ea11af053b7e5d964..86d0080f8e658ad687baf028bb1329d62b965848 100644
--- a/noise_cifar_train.py
+++ b/noise_cifar_train.py
@@ -11,7 +11,7 @@ import torch.backends.cudnn as cudnn
 from utils import train_noise, test, get_output, WeightEMA
 from dataset import get_cifar_dataset
 from networks.wideresnet import Wide_ResNet
-from networks.cnn import CNN #Import custom CNN
+#from networks.cnn import CNN #Import custom CNN
 from networks.convnet import ConvNet
 
 def log(path, str):