pointnet2 working - TODO: Eval!

main.py

@@ -5,12 +5,11 @@ import warnings
 
 import torch
 from pytorch_lightning import Trainer
-from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
+from pytorch_lightning.callbacks import ModelCheckpoint  # , EarlyStopping
 
 from ml_lib.modules.util import LightningBaseModule
-from ml_lib.utils.config import Config
 from ml_lib.utils.logging import Logger
-from ml_lib.utils.model_io import SavedLightningModels
+from utils.project_config import ThisConfig
 
 warnings.filterwarnings('ignore', category=FutureWarning)
 warnings.filterwarnings('ignore', category=UserWarning)
@@ -33,11 +32,13 @@ def run_lightning_loop(config_obj):
         # =============================================================================
         # Early Stopping
         # TODO: For This to work, one must set a validation step and End Eval and Score
+        """
         early_stopping_callback = EarlyStopping(
             monitor='val_loss',
             min_delta=0.0,
             patience=0,
         )
+        """
 
         # Model
         # =============================================================================
@@ -76,6 +77,9 @@ def run_lightning_loop(config_obj):
 
 
 if __name__ == "__main__":
-    from ._parameters import args
-    config = Config.read_namespace(args)
+    from _parameters import args
+    from ml_lib.utils.tools import fix_all_random_seeds
+
+    config = ThisConfig.read_namespace(args)
+    fix_all_random_seeds(config)
     trained_model = run_lightning_loop(config)