Refactor config and trainer scripts to rename get_existing_rounds method to __get_existing_rounds and update model round initialization logic

Author: DefinetlyNotAI

v4n1_Trainer.py

@@ -124,7 +124,7 @@ def train(config: TrainingConfig, resources: dict):
 
     # ----------------- RUN ------------------
     available_dataset = [10, 100, 1000, 5000, 10000, 17500, 25000]
-    for dataset in available_dataset:
+    for loop_idx, dataset in enumerate(available_dataset, start=1):
         if dataset <= 1000:
             name = "SenseNano"
         elif 1000 < dataset <= 5000:
@@ -133,8 +133,7 @@ def train(config: TrainingConfig, resources: dict):
             name = "Sense"
         else:
             name = "SenseMacro"
-        existing_rounds = cfg.get_existing_rounds(cfg.CACHE_DIR)  # Auto-increment round based on existing folders
-        model_round = max(existing_rounds) + 1 if existing_rounds else 1
+        model_round = loop_idx
         cfg.update({
             # Model / caching / logging
             "MODEL_NAME": f"Model_{name}.4n1",  # Name of the model for identification and caching

vulnscan/config.py

@@ -9,7 +9,7 @@ def __init__(self, model_name: str = "Model_Sense.4n1"):
         self.MODEL_NAME = model_name
         self.CACHE_DIR = os.path.join(os.getcwd(), "cache")
 
-        existing_rounds = self.get_existing_rounds(self.CACHE_DIR)  # Auto-increment round based on existing folders
+        existing_rounds = self.__get_existing_rounds(self.CACHE_DIR)  # Auto-increment round based on existing folders
         self.MODEL_ROUND = max(existing_rounds) + 1 if existing_rounds else 1
 
         self.LOG_FILE = f"{self.CACHE_DIR}/{self.MODEL_NAME}/training.log"
@@ -59,7 +59,7 @@ def __init__(self, model_name: str = "Model_Sense.4n1"):
         os.makedirs(self.DATA_CACHE_DIR, exist_ok=True)
 
     @staticmethod
-    def get_existing_rounds(cache_dir: str) -> list[int]:
+    def __get_existing_rounds(cache_dir: str) -> list[int]:
         """
         Returns a list of round numbers based on existing folders in the cache directory.
         """