Update README.md

Author: calmiLovesAI

README.md

@@ -7,6 +7,7 @@ A tensorflow2 implementation of some basic CNNs.
 + [MobileNet_V3](https://github.com/calmisential/MobileNetV3_TensorFlow2)
 + [EfficientNet](https://github.com/calmisential/EfficientNet_TensorFlow2)
 + [ResNeXt](https://github.com/calmisential/ResNeXt_TensorFlow2)
++ [InceptionV4, InceptionResNetV1, InceptionResNetV2](https://github.com/calmisential/InceptionV4_TensorFlow2)
 
 ## Other networks
 For AlexNet and VGG, see : https://github.com/calmisential/TensorFlow2.0_Image_Classification<br/>

train.py

@@ -62,14 +62,14 @@ def process_features(features):
 
     return images, labels
 
+
 if __name__ == '__main__':
     # GPU settings
     gpus = tf.config.experimental.list_physical_devices('GPU')
     if gpus:
         for gpu in gpus:
             tf.config.experimental.set_memory_growth(gpu, True)
 
-
     # get the dataset
     train_dataset, valid_dataset, test_dataset, train_count, valid_count, test_count = generate_datasets()
 
@@ -120,7 +120,6 @@ def valid_step(image_batch, label_batch):
                                                                                      train_loss.result().numpy(),
                                                                                      train_accuracy.result().numpy()))
 
-
         for features in valid_dataset:
             valid_images, valid_labels = process_features(features)
             valid_step(valid_images, valid_labels)