clean useless-code

dataset_checker.py

@@ -1,7 +1,4 @@
 import cv2
-import numpy as np
-import tensorflow as tf
-import tensorflow_datasets as tfds
 from modules.utils import set_memory_growth
 from modules.dataset import load_cifar10_dataset
 

modules/lr_scheduler.py

@@ -1,5 +1,4 @@
 import tensorflow as tf
-import math
 
 
 def MultiStepLR(initial_learning_rate, lr_steps, lr_rate, name='MultiStepLR'):

modules/models.py

@@ -1,8 +1,7 @@
-import functools
 import tensorflow as tf
 from absl import logging
 from tensorflow.keras import Model, Sequential
-from tensorflow.keras.layers import (Input, Dense, Flatten, Dropout, Conv2D,
+from tensorflow.keras.layers import (Input, Dense, Flatten, Conv2D,
                                      AveragePooling2D, GlobalAveragePooling2D,
                                      ReLU)
 from modules.operations import (OPS, FactorizedReduce, ReLUConvBN,

modules/models_search.py

@@ -1,13 +1,10 @@
-import functools
 import tensorflow as tf
 from absl import logging
 from tensorflow.keras import Model, Sequential
-from tensorflow.keras.layers import (Input, Dense, Flatten, Dropout, Conv2D,
-                                     MaxPool2D, GlobalAveragePooling2D,
-                                     ReLU, Softmax)
+from tensorflow.keras.layers import (Input, Dense, Flatten, Conv2D, MaxPool2D,
+                                     GlobalAveragePooling2D, Softmax)
 from modules.operations import (OPS, FactorizedReduce, ReLUConvBN,
-                                BatchNormalization, Identity, drop_path,
-                                kernel_init, regularizer)
+                                BatchNormalization, kernel_init, regularizer)
 from modules.genotypes import PRIMITIVES, Genotype
 
 

modules/operations.py

@@ -1,7 +1,7 @@
 import tensorflow as tf
 from tensorflow.keras import Sequential
 from tensorflow.keras.layers import (Conv2D, SeparableConv2D, MaxPool2D,
-                                     AveragePooling2D, ReLU, Dropout)
+                                     AveragePooling2D, ReLU)
 
 
 OPS = {'none': lambda f, s, wd, affine: Zero(s),

modules/utils.py

@@ -1,4 +1,3 @@
-import cv2
 import yaml
 import sys
 import time

train.py

@@ -1,7 +1,6 @@
 from absl import app, flags, logging
 from absl.flags import FLAGS
 import os
-import numpy as np
 import tensorflow as tf
 
 from modules.models import CifarModel

train_search.py

@@ -1,7 +1,6 @@
 from absl import app, flags, logging
 from absl.flags import FLAGS
 import os
-import numpy as np
 import tensorflow as tf
 
 from modules.models_search import SearchNetArch