文件名称：pytorch-benchmarks:Pytorch的基准框架
文件大小：12KB
文件格式：ZIP
更新时间：2024-04-26 04:51:14
Python
PyTorch基准 使用PyTorch对多GPU系统进行基准测试的工具。 用法 运行main.py以开始基准测试！ 可选参数： --warm_up_steps，-w：每个时期的预热步骤数。 预热步骤将不予考虑； 默认值：10 --num_epochs，-n：纪元数; 默认值：10 --batch_size，-b：批处理大小; 默认值：64 --num_gpus，-g：用于训练的gpu数量; 默认值：1 --gpu_ids，-i：用于训练的GPU的ID。 如果未给出，则使用range（num_gpus-1）。 --model，-m：用于训练的模型; 默认值：resnet50 --use_fp16，-f：使用半精度; 默认值：False --img_folder，-imf：训练图像的目的地。 如果未给出，则使用随机数据。 --num_workers，-nw：数据加载器使用的工作程序数。 如果
【文件预览】：
pytorch-benchmarks-main
----model_utils.py(6KB)
----flags.py(4KB)
----data.py(4KB)
----utils.py(10KB)
----main.py(4KB)
----requirements.txt(50B)
----LICENSE(1KB)
----README.md(832B)
----.gitignore(106B)
----gpu_live_plot.py(4KB)