Change plot settings

mosaicml · Jan 23, 2024 · 808ced5 · 808ced5
1 parent 1764b72
commit 808ced5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llmfoundry/utils/validation_utils.py b/llmfoundry/utils/validation_utils.py
@@ -814,8 +814,8 @@ def plot_hist(data, save_plot_path=None):
 
     # Aesthetics
     plt.title('Histogram of Token Counts')
-    plt.xlabel('Token Count')
-    plt.ylabel('Frequency')
+    plt.xlabel('Number of Tokens per Sample')
+    plt.ylabel('Count of Frequency')
 
     # Grid and Layout
     plt.grid(axis='y', alpha=0.75)