Update required tensorflow version

Author: calmiLovesAI

README.md

@@ -22,7 +22,7 @@ For ResNet, see : https://github.com/calmisential/TensorFlow2.0_ResNet
 ## Train
 1. Requirements:
 + Python >= 3.6
-+ Tensorflow == 2.0.0
++ Tensorflow == 2.1.0
 2. To train the network on your own dataset, you can put the dataset under the folder **original dataset**, and the directory should look like this:
 ```
 |——original dataset

train.py

@@ -101,7 +101,7 @@ def process_features(features):
 
 if __name__ == '__main__':
     # GPU settings
-    gpus = tf.config.experimental.list_physical_devices('GPU')
+    gpus = tf.config.list_physical_devices("GPU")
     if gpus:
         for gpu in gpus:
             tf.config.experimental.set_memory_growth(gpu, True)