From 8281331e724ac0af8ccb8de58ba82ee99281480c Mon Sep 17 00:00:00 2001
From: namsaraeva <a.namsaraeva@tum.de>
Date: Fri, 17 May 2024 12:55:21 +0200
Subject: [PATCH] add comment

---
 src/sparcscore/ml/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py
index 80d9fda..0af6f26 100644
--- a/src/sparcscore/ml/models.py
+++ b/src/sparcscore/ml/models.py
@@ -72,7 +72,7 @@ def make_layers_MLP(self, cfg_MLP, cfg, regression = False):
         # get output feature size of CNN with chosen configuration
         
         if regression:
-            in_features = int(cfg[-2]) * 4 * 4
+            in_features = int(cfg[-2]) * 4 * 4 # needed for the STING screen since the images are big
         else:
             in_features = int(cfg[-2]) * 2 * 2