[DRAFT] add examples of scalability plots from virgo

docs/tutorials/distrib-ml/torch_scaling_test.rst

@@ -6,6 +6,41 @@ PyTorch scaling test
    :end-before: Below follows an example of
 
 
-Below follows an example of scalability plot generated by ``itwinai scalability-report``:
+Plots of the scalability metrics
+-----------------------------
 
-.. image:: ../../../tutorials/distributed-ml/torch-scaling-test/img/report.png
+We have the following scalability metrics available: 
+
+- Absolute wall-clock time comparison
+- Relative wall-clock time speedup
+- Communication vs. Computation time
+- GPU Utilization (%)
+- Power Consumption (Watt)
+
+Some examples of these scalability metrics on the Virgo use case with one, two and four
+nodes respectively can be seen below: 
+
+Absolute Wall-Clock Time Comparison
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. image:: ../../../tutorials/distributed-ml/torch-scaling-test/img/absolute-time.png
+
+Relative Wall-Clock Time Speedup
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. image:: ../../../tutorials/distributed-ml/torch-scaling-test/img/relative-speedup.png
+
+Communication vs Computation
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. image:: ../../../tutorials/distributed-ml/torch-scaling-test/img/comp-vs-comm.png
+
+GPU Utilization
+~~~~~~~~~~~~~~~~
+
+.. image:: ../../../tutorials/distributed-ml/torch-scaling-test/img/gpu-utilization.png
+
+Power Consumption
+~~~~~~~~~~~~~~~~~
+
+.. image:: ../../../tutorials/distributed-ml/torch-scaling-test/img/energy-consumption.png

tutorials/distributed-ml/torch-scaling-test/README.md

@@ -137,7 +137,3 @@ To see the full list of possible arguments, type:
 ```bash
 itwinai generate-scalability-plot --help
 ```
-
-Below follows an example of scalability plot generated by `itwinai scalability-report`:
-
-![report](img/report.png)