add _data_pipeline back

Lightning-Universe · Mar 29, 2021 · e2f24dc · e2f24dc
1 parent 0245d17
commit e2f24dc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flash/core/model.py b/flash/core/model.py
@@ -78,6 +78,7 @@ def __init__(
         # TODO: should we save more? Bug on some regarding yaml if we save metrics
         self.save_hyperparameters("learning_rate", "optimizer")
 
+        self._data_pipeline = None
         self._preprocess = None
         self._postprocess = None