Add warning for missing val_loss

MALES-project · Nov 3, 2024 · e73cc0b · e73cc0b
1 parent 27d0dd4
commit e73cc0b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/speckcn2/loss.py b/src/speckcn2/loss.py
@@ -57,7 +57,11 @@ def __init__(self,
                  validation: bool = False):
         super(ComposableLoss, self).__init__()
         if validation:
-            config['loss'] = config['val_loss']
+            if 'val_loss' in config:
+                config['loss'] = config['val_loss']
+            else:
+                print('[!] Warning: Validation loss not found in config.yaml,',
+                      'keeping track of training loss instead')
         self.device = device
         self.loss_functions: dict[str, Callable] = {
             'MSE': torch.nn.MSELoss(),