README.md

# 介绍

 该测试用例可用于ResNet50等网络在DTK平台的性能测试及精度验证。

# 单卡测试（单精度）

## 运行

  export PYTHONPATH=$PYTHONPATH:/path/to/tensorflow/model
  export HIP_VISIBLE_DEVICES=0
  python3 official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py --data_dir=/path/to/{ImageNet-tensorflow_data_dir} --model_dir=/path/to/{model_save_dir} --batch_size=128 --num_gpus=1  --use_synthetic_data=false
  
# 单卡测试 （混合精度）

## 运行 

  export PYTHONPATH=$PYTHONPATH:/path/to/tensorflow/model
  export HIP_VISIBLE_DEVICES=0
  python3 official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py --data_dir=/path/to/{ImageNet-tensorflow_data_dir} --model_dir=/path/to/{model_save_dir} --batch_size=128 --num_gpus=1  --use_synthetic_data=false --dtype=fp16

# 多卡测试 （单精度）

## 运行

  mpirun -np $np --hostfile hostfile --bind-to none scripts-run/single_process.sh

  多卡测试时需要修改部分代码，具体可参考[https://tensorflow.google.cn/guide/migrate/multi_worker_cpu_gpu_training?hl=en]   

# 参考
[https://github.com/tensorflow/models/tree/v2.7.0](https://github.com/tensorflow/models/tree/v2.7.0/official/vision/image_classification/resnet)