added comment

mlcommons · Jul 19, 2023 · fc36c7d · fc36c7d
1 parent 059c4a3
commit fc36c7d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gandlf_collectStats b/gandlf_collectStats
@@ -14,6 +14,7 @@ from GANDLF.cli import copyrightMessage
 def plot_all(df_training, df_validation, df_testing, output_plot_dir):
     """
     Plots training, validation, and testing data for loss and other metrics.
+    TODO: this function needs to be moved under utils and then called after every training epoch.
 
     Args:
         df_training (pd.DataFrame): DataFrame containing training data.