added warning on zero initialization (#74)

noegroup · Mar 24, 2023 · fbba56f · fbba56f
1 parent 03d8e19
commit fbba56f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bgflow/factory/generator_builder.py b/bgflow/factory/generator_builder.py
@@ -173,6 +173,7 @@ def build_flow(self, zero_parameters=False):
         """
         flow = SequentialFlow(self.layers)
         if zero_parameters:
+            warnings.warn("Initializing the flow with zeros makes it much less flexible", UserWarning)
             for p in flow.parameters():
                 p.data.zero_()
         return flow