bug fix

allenai · Nov 14, 2024 · 662db73 · 662db73
1 parent 7edf3ca
commit 662db73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/scaling/stacked.py b/scripts/scaling/stacked.py
@@ -80,7 +80,7 @@ def main():
 
         # keys = task.get_loss_keys()
         # data_by_name = get_final_data_by_name(configs, keys, num_to_avg=args.num_to_avg)
-        data_by_name = get_downstream_data_by_name(configs, task_name, moving_avg=args.num_to_avg, last_n_points=1)
+        data_by_name = get_downstream_data_by_name(configs, task_name, moving_avg=args.moving_avg, last_n_points=1)
 
         train_nds, train_xs = [], []
         for name, data in data_by_name.items():