"alphafold/model/data.py" did not exist on "2f0d89e765051fc9e26fb4c52e5ad91bbb0e7e0b"
README.md 1.24 KB
Newer Older
qianyj's avatar
qianyj committed
1
# 介绍
qianyj's avatar
qianyj committed
2
3
4
5
6
7
8
9
10
11

 该测试用例可用于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
qianyj's avatar
qianyj committed
12
  
qianyj's avatar
qianyj committed
13
# 单卡测试 (混合精度)
qianyj's avatar
qianyj committed
14

qianyj's avatar
qianyj committed
15
## 运行 
qianyj's avatar
qianyj committed
16

qianyj's avatar
qianyj committed
17
18
19
  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
qianyj's avatar
qianyj committed
20

qianyj's avatar
qianyj committed
21
22
23
24
25
26
27
# 多卡测试 (单精度)

## 运行

  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]   
qianyj's avatar
qianyj committed
28
29
30

# 参考
[https://github.com/tensorflow/models/tree/v2.7.0](https://github.com/tensorflow/models/tree/v2.7.0/official/vision/image_classification/resnet)
qianyj's avatar
qianyj committed
31