print info

GreenWizard2015 · Apr 10, 2024 · 8590866 · 8590866
1 parent cabb839
commit 8590866
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/NN/encoders/Encoder.py b/NN/encoders/Encoder.py
@@ -163,7 +163,11 @@ def _createGlobalContextModel(X, config, latentDim, name):
   raise NotImplementedError('Unknown global context model: {}'.format(model))
 
 def _withPositionConfig(config, name):
-  if config is None: return lambda x, _: x
+  if config is None:
+    print('[Encoder] Positions: No')
+    return lambda x, _: x
+
+  print('[Encoder] Positions: Yes')
 
   if isinstance(config, bool): config = { 'N': 32 }
   assert isinstance(config, dict), 'config must be a dictionary'

diff --git a/NN/restorators/IRestorationProcess.py b/NN/restorators/IRestorationProcess.py
@@ -20,6 +20,8 @@ def __init__(self, predictions):
     if 'rgb' not in predictions: predictions.insert(0, 'rgb')
     self._outputs = predictions
     self._channels = IRestorationProcess.getOutputSize(predictions)
+
+    print('[IRestorationProcess] Restorator:', self.__class__.__name__)
     return
 
   def forward(self, x0, xT=None):