update TensorFlow test code

TensorFlow2x/Accuracy_Validation/ResNet50_Official/README.md

@@ -23,7 +23,7 @@
 ## 运行
  
     # sed指令只需要执行一次，添加支持多卡运行的代码
-    sed -i '99 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
+    sed -i '101 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
     
     export PYTHONPATH=/path/to/tensorflow/model:$PYTHONPATH
     mpirun -np ${num_gpu}  --hostfile hostfile  -mca btl self,tcp  --bind-to none scripts-run/single_process.sh
@@ -33,7 +33,7 @@
 ## 运行
    
     # sed指令只需要执行一次，添加支持多卡运行的代码
-    sed -i '99 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
+    sed -i '101 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
     
     修改scripts-run/single_process.sh中的--dtype=fp16
     export PYTHONPATH=/path/to/tensorflow/model:$PYTHONPATH

TensorFlow2x/Accuracy_Validation/ResNet50_Official/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py

@@ -16,6 +16,7 @@
 
 import math
 import os
+import json
 
 # Import libraries
 from absl import app
@@ -97,6 +98,7 @@ def run(flags_obj):
   Returns:
     Dictionary of training and eval stats.
   """
+  
   keras_utils.set_session_config()
   performance.set_mixed_precision_policy(flags_core.get_tf_dtype(flags_obj))
 

TensorFlow2x/ComputeVision/Classification/README.md

@@ -19,7 +19,7 @@
 ## 分布式多卡
     
     # sed指令只需要执行一次，添加支持多卡运行的代码
-    sed -i '99 r configfile' models-master/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
+    sed -i '101 r configfile' models-master/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
     
     export PYTHONPATH=/path/to/tensorflow/model:$PYTHONPATH
     mpirun -np ${num_gpu} --hostfile hostfile  -mca btl self,tcp  --bind-to none scripts-run/single_process.sh

TensorFlow2x/ComputeVision/Classification/models-master/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py

@@ -17,6 +17,7 @@
 import math
 import os
 import json
+
 # Import libraries
 from absl import app
 from absl import flags
@@ -97,6 +98,7 @@ def run(flags_obj):
   Returns:
     Dictionary of training and eval stats.
   """
+  
   keras_utils.set_session_config()
   performance.set_mixed_precision_policy(flags_core.get_tf_dtype(flags_obj))