print info

Author: GreenWizard2015

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'

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):