Merge pull request #2599 from fongjiantan/fix/ONSAM-2083

[ONSAM-2083]Fix for IndexError

Author: jimmytwei

Diff for: 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("Incorrect data size to calculate speedup")