Update noise_cifar_train.py

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):