文件名称:ml-performance-benchmark:MLAI工作负载的性能基准测试
文件大小:13.08MB
文件格式:ZIP
更新时间:2024-04-28 08:24:52
Python
毫升性能基准 ML / AI工作负载的性能基准测试。 ResNet Imagenet数据:从下载培训和验证数据集 使用以下命令解压缩数据: tar xf ILSVRC2012_img_val.tar -C $IMAGENET_DATA_HOME/validation tar xf ILSVRC2012_img_train.tar -C $IMAGENET_DATA_HOME/train 数据必须转换为TFRecords格式; 这可以通过脚本来完成 python imagenet_to_gcs.py \ --raw_data_dir=$IMAGENET_DATA_HOME \ --local_scratch_dir=$IMAGENET_DATA_HOME/tf_records \ --nogcs_upload Resnet目录基于: ://code.ornl.gov/olc