Unverified Commit f283b536 authored by Yuting Jiang's avatar Yuting Jiang Committed by GitHub
Browse files

Config - Disable disk-benchmark in ndmv4.yaml and change batch size to 1 in default.yaml (#292)

**Description**
Disable disk-benchmark in ndmv4.yaml and change batch size to 1 in default.yaml
parent 7d7cd3dc
...@@ -81,7 +81,7 @@ superbench: ...@@ -81,7 +81,7 @@ superbench:
prefix: CUDA_VISIBLE_DEVICES={proc_rank} numactl -N $(({proc_rank}/2)) prefix: CUDA_VISIBLE_DEVICES={proc_rank} numactl -N $(({proc_rank}/2))
parallel: no parallel: no
disk-benchmark: disk-benchmark:
enable: true enable: false
modes: modes:
- name: local - name: local
proc_num: 1 proc_num: 1
......
...@@ -83,7 +83,21 @@ superbench: ...@@ -83,7 +83,21 @@ superbench:
proc_num: 1 proc_num: 1
parallel: no parallel: no
parameters: parameters:
block_devices: [] block_devices:
- /dev/nvme0n1
- /dev/nvme1n1
- /dev/nvme2n1
- /dev/nvme3n1
- /dev/nvme4n1
- /dev/nvme5n1
- /dev/nvme6n1
- /dev/nvme7n1
seq_read_runtime: 60
seq_write_runtime: 60
seq_readwrite_runtime: 60
rand_read_runtime: 60
rand_write_runtime: 60
rand_readwrite_runtime: 60
gpu-copy-bw: gpu-copy-bw:
enable: true enable: true
modes: modes:
......
...@@ -26,6 +26,7 @@ superbench: ...@@ -26,6 +26,7 @@ superbench:
duration: 0 duration: 0
num_warmup: 16 num_warmup: 16
num_steps: 128 num_steps: 128
batch_size: 1
precision: precision:
- float32 - float32
- float16 - float16
...@@ -143,6 +144,8 @@ superbench: ...@@ -143,6 +144,8 @@ superbench:
port: 22 port: 22
ort-inference: ort-inference:
<<: *default_local_mode <<: *default_local_mode
parameters:
batch_size: 1
tensorrt-inference: tensorrt-inference:
<<: *default_local_mode <<: *default_local_mode
parameters: parameters:
...@@ -155,7 +158,7 @@ superbench: ...@@ -155,7 +158,7 @@ superbench:
- bert-base - bert-base
- bert-large - bert-large
seq_length: 224 seq_length: 224
batch_size: 32 batch_size: 1
precision: int8 precision: int8
gpt_models: gpt_models:
<<: *default_pytorch_mode <<: *default_pytorch_mode
...@@ -164,7 +167,6 @@ superbench: ...@@ -164,7 +167,6 @@ superbench:
- gpt2-large - gpt2-large
parameters: parameters:
<<: *common_model_config <<: *common_model_config
batch_size: 4
bert_models: bert_models:
<<: *default_pytorch_mode <<: *default_pytorch_mode
models: models:
...@@ -172,14 +174,12 @@ superbench: ...@@ -172,14 +174,12 @@ superbench:
- bert-large - bert-large
parameters: parameters:
<<: *common_model_config <<: *common_model_config
batch_size: 8
lstm_models: lstm_models:
<<: *default_pytorch_mode <<: *default_pytorch_mode
models: models:
- lstm - lstm
parameters: parameters:
<<: *common_model_config <<: *common_model_config
batch_size: 128
resnet_models: resnet_models:
<<: *default_pytorch_mode <<: *default_pytorch_mode
models: models:
...@@ -188,7 +188,6 @@ superbench: ...@@ -188,7 +188,6 @@ superbench:
- resnet152 - resnet152
parameters: parameters:
<<: *common_model_config <<: *common_model_config
batch_size: 128
densenet_models: densenet_models:
<<: *default_pytorch_mode <<: *default_pytorch_mode
models: models:
...@@ -196,7 +195,6 @@ superbench: ...@@ -196,7 +195,6 @@ superbench:
- densenet201 - densenet201
parameters: parameters:
<<: *common_model_config <<: *common_model_config
batch_size: 128
vgg_models: vgg_models:
<<: *default_pytorch_mode <<: *default_pytorch_mode
models: models:
...@@ -206,4 +204,3 @@ superbench: ...@@ -206,4 +204,3 @@ superbench:
- vgg19 - vgg19
parameters: parameters:
<<: *common_model_config <<: *common_model_config
batch_size: 128
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment