Fix PyTorch training benchmark doc template (#5357)

* fix template

* update wordlist

.wordlist.txt

@@ -622,6 +622,7 @@ coalescable
 codename
 collater
 comgr
+compat
 completers
 composable
 concretization
@@ -783,6 +784,7 @@ lossy
 macOS
 matchers
 maxtext
+megablocks
 megatron
 microarchitecture
 migraphx
@@ -941,6 +943,7 @@ softmax
 spack
 spmm
 src
+stanford
 stochastically
 strided
 subcommand

docs/how-to/rocm-for-ai/training/benchmark-docker/pytorch-training.rst

@@ -164,20 +164,20 @@ Run training
    {% for model_group in model_groups %}
       {% for model in model_group.models %}
 
-         The following run command is tailored to {{ model.model }}.
-         See :ref:`amd-pytorch-training-model-support` to switch to another available model.
-
-         1. Clone the ROCm Model Automation and Dashboarding (`<https://github.com/ROCm/MAD>`__) repository to a local
-            directory and install the required packages on the host machine.
-
-            .. code-block:: shell
-
-               git clone https://github.com/ROCm/MAD
-               cd MAD
-               pip install -r requirements.txt
-
          .. container:: model-doc {{ model.mad_tag }}
 
+            The following run command is tailored to {{ model.model }}.
+            See :ref:`amd-pytorch-training-model-support` to switch to another available model.
+
+            1. Clone the ROCm Model Automation and Dashboarding (`<https://github.com/ROCm/MAD>`__) repository to a local
+               directory and install the required packages on the host machine.
+
+               .. code-block:: shell
+
+                  git clone https://github.com/ROCm/MAD
+                  cd MAD
+                  pip install -r requirements.txt
+
             2. For example, use this command to run the performance benchmark test on the {{ model.model }} model
                using one node with the {{ model.precision }} data type on the host machine.
 
@@ -202,8 +202,10 @@ Run training
    {% for model_group in model_groups %}
       {% for model in model_group.models %}
 
-         The following commands are tailored to {{ model.model }}.
-         See :ref:`amd-pytorch-training-model-support` to switch to another available model.
+         .. container:: model-doc {{ model.mad_tag }}
+
+            The following commands are tailored to {{ model.model }}.
+            See :ref:`amd-pytorch-training-model-support` to switch to another available model.
 
       {% endfor %}
    {% endfor %}