Update nn_transformer_regressor.py

bartzbeielstein · bartzbeielstein · commit ba85cce6068e · 2024-09-22T16:01:03.000+02:00
diff --git a/src/spotpython/light/regression/nn_transformer_regressor.py b/src/spotpython/light/regression/nn_transformer_regressor.py
@@ -32,6 +32,7 @@ def __init__(
 
         self.d_model = d_model_mult * nhead
         self.dim_feedforward = dim_feedforward_mult * self.d_model
+        print(f"d_model: {self.d_model}, dim_feedforward: {self.dim_feedforward}")
 
         if _torchmetric is None:
             _torchmetric = "mean_squared_error"
