[ONSAM-2083]Fix for IndexError

This fixes IndexError in IntelTensorFlow_Enabling_Auto_Mixed_Precision_for_TransferLearning

Before this fix, throughput_list could be less than 2 elements and causing IndexError. With the added check, speedup will only be calculated if there are at least two elements.

Fixes [ONSAM-2083]

Signed-off-by: syeo <[email protected]>

Author: rayyeo3006

AI-and-Analytics/Features-and-Functionality/IntelTensorFlow_Enabling_Auto_Mixed_Precision_for_TransferLearning/scripts/plot.py

@@ -8,8 +8,10 @@
         throughput = line.split(': ')[1]
         throughput_list.append(float(throughput))
 
-print("Throughput list: ", throughput_list)
-speedup = float(throughput_list[1])/float(throughput_list[0])
-print("Speedup : ", speedup)
-df = pd.DataFrame({'pretrained_model':['saved model', 'optimized model'], 'Speedup':[1, speedup]})
-ax = df.plot.bar( x='pretrained_model', y='Speedup', rot=0)
+if len(throughput_list) >= 2:
+    speedup = float(throughput_list[1])/float(throughput_list[0])
+    print("Speedup : ", speedup)
+    df = pd.DataFrame({'pretrained_model':['saved model', 'optimized model'], 'Speedup':[1, speedup]})
+    ax = df.plot.bar( x='pretrained_model', y='Speedup', rot=0)
+else:
+    print("Insufficient data to calculate speedup")