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