Clarify override comment

datamol-io · Oct 26, 2023 · cc6adba · cc6adba
1 parent 6832a09
commit cc6adba
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/graphium/cli/train_finetune_test.py b/graphium/cli/train_finetune_test.py
@@ -72,6 +72,7 @@ def get_replication_factor(cfg):
 def get_gradient_accumulation_factor(cfg):
     """
     WARNING: This MUST be called after accelerator overrides have been applied
+    (i.e. after `load_accelerator` has been called)
     """
     try:
         # Navigate through the nested dictionaries and get the gradient accumulation factor
@@ -89,6 +90,7 @@ def get_gradient_accumulation_factor(cfg):
 def get_training_batch_size(cfg):
     """
     WARNING: This MUST be called after accelerator overrides have been applied
+    (i.e. after `load_accelerator` has been called)
     """
     try:
         # Navigate through the nested dictionaries and get the training batch size