moved esm2 scheduler into llm

NVIDIA · Nov 5, 2024 · 79381a9 · 79381a9
1 parent bfcb2c9
commit 79381a9
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 7 deletions.
diff --git a/...m2/src/bionemo/esm2/model/lr_scheduler.py → ...llm/src/bionemo/llm/model/lr_scheduler.py b/...m2/src/bionemo/esm2/model/lr_scheduler.py → ...llm/src/bionemo/llm/model/lr_scheduler.py
diff --git a/sub-packages/bionemo-llm/src/bionemo/llm/train.py b/sub-packages/bionemo-llm/src/bionemo/llm/train.py
@@ -126,13 +126,6 @@ def setup_trainer(
             )
         )
 
-    # TODO set these as flags, the following are needed:
-    """
-    nsys_profiling (bool)
-    nsys_start_step (int) when to start profiling
-    nsys_end_step (int) when to stop profiling
-    nsys_ranks (List[int]) which ranks to profile.
-    """
     if nsys_config:
         if nsys_config.end_step is None:
             nsys_config.end_step = training_config.max_steps