Pytorch-benchmarks是一款用于衡量Pytorch性能的基准框架

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：数据加载器使用的工作程序数。 如果