1 Star 0 Fork 1

NickTan/gpu-perf

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
gpu_perf.sh 790 Bytes
一键复制 编辑 原始数据 按行查看 历史
july_carbon 提交于 2023-07-14 16:59 . Add yolov5nu yolov8n in gpu_perf.sh
#! /bin/bash
source ${HOME}/miniconda3/etc/profile.d/conda.sh
conda activate gpu-perf
for dataset in fake byModel; do
for model in resnet50 resnet101 vit_b_16 yolov5nu yolov8n; do
for precision in fp32 fp16; do
for batch_size in 1 10 32 64 128; do
python gpu_perf.py --mode eval --model-type ${model} --batch-size ${batch_size} --${precision} --dataset ${dataset}
done
done
done
for model in resnet50 resnet101 vit_b_16 yolov5nu yolov8n; do
for precision in fp32 mixed; do
for batch_size in 16 32 64 128; do
python gpu_perf.py --mode train --model-type ${model} --batch-size ${batch_size} --${precision} --dataset ${dataset}
done
done
done
done
conda deactivate
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/tgg_920721/gpu-perf.git
git@gitee.com:tgg_920721/gpu-perf.git
tgg_920721
gpu-perf
gpu-perf
master

搜索帮助