djstrong commited on
Commit
f820897
·
1 Parent(s): 32bb346
Files changed (1) hide show
  1. plot_results.py +6 -0
plot_results.py CHANGED
@@ -133,6 +133,12 @@ def create_performance_plot(csv_path='benchmark_results.csv', metadata_path='met
133
  frontier_x, frontier_y = zip(*frontier_points)
134
  plt.plot(frontier_x, frontier_y, 'r--', label='Pareto frontier')
135
 
 
 
 
 
 
 
136
  # Customize the plot
137
  plt.grid(True, linestyle='--', alpha=0.7)
138
  plt.xlabel('Model Size (billions of parameters)')
 
133
  frontier_x, frontier_y = zip(*frontier_points)
134
  plt.plot(frontier_x, frontier_y, 'r--', label='Pareto frontier')
135
 
136
+ # Add vertical line for consumer GPU budget
137
+ plt.axvline(x=12, color='gray', linestyle=':', label='Consumer-budget GPU limit', ymin=-0.15, clip_on=False)
138
+ plt.text(12, -0.15, 'Consumer-budget\nGPU (24GB) limit\nin full precision',
139
+ horizontalalignment='center', verticalalignment='top',
140
+ transform=plt.gca().get_xaxis_transform())
141
+
142
  # Customize the plot
143
  plt.grid(True, linestyle='--', alpha=0.7)
144
  plt.xlabel('Model Size (billions of parameters)')