Update doc/train/parallel-training.md

Co-authored-by: Jinzhe Zeng <[email protected]>
Signed-off-by: Lysithea <[email protected]>

doc/train/parallel-training.md

@@ -4,7 +4,7 @@
 **Supported backends**: TensorFlow {{ tensorflow_icon }}, PyTorch {{ pytorch_icon }}
 :::
 
-## TensorFlow Implementation
+## TensorFlow Implementation {{ tensorflow_icon }}
 Currently, parallel training in tensorflow version is enabled in a synchronized way with help of [Horovod](https://github.com/horovod/horovod).
 Depending on the number of training processes (according to MPI context) and the number of GPU cards available, DeePMD-kit will decide whether to launch the training in parallel (distributed) mode or in serial mode. Therefore, no additional options are specified in your JSON/YAML input file.