Benchmarks: Configuration - Add gpt-small into config files. (#253)

**Description** Add gpt-small into config files.

Benchmarks: Configuration - Add gpt-small into config files. (#253)
**Description** Add gpt-small into config files.
8042fa34 · guoshzhao · GitHub · 371fd61c · 8042fa34 · 8042fa34
Unverified Commit 8042fa34 authored Dec 02, 2021 by guoshzhao Committed by GitHub Dec 02, 2021
3 changed files
--- a/superbench/config/amd_mi100_hpe.yaml
+++ b/superbench/config/amd_mi100_hpe.yaml
@@ -92,6 +92,7 @@ superbench:
    gpt_models:
      <<: *default_pytorch_mode
      models:
+        - gpt2-small
        - gpt2-large
      parameters:
        <<: *common_model_config

--- a/superbench/config/amd_mi100_z53.yaml
+++ b/superbench/config/amd_mi100_z53.yaml
@@ -93,6 +93,7 @@ superbench:
    gpt_models:
      <<: *default_pytorch_mode
      models:
+        - gpt2-small
        - gpt2-large
      parameters:
        <<: *common_model_config

--- a/superbench/config/azure_ndv4.yaml
+++ b/superbench/config/azure_ndv4.yaml
@@ -97,6 +97,7 @@ superbench:
    gpt_models:
      <<: *default_pytorch_mode
      models:
+        - gpt2-small
        - gpt2-large
      parameters:
        <<: *common_model_config