diff --git a/predtuner/modeledapp.py b/predtuner/modeledapp.py index 06efc8cc69e7898b667f775bb68bb1a5be60a269..680c5f195803ef7d5df6f9da253ebdfaa1854111 100644 --- a/predtuner/modeledapp.py +++ b/predtuner/modeledapp.py @@ -200,7 +200,7 @@ class QoSModelP1(IQoSModel): self.delta_tensors = { op: {k.name: None for k in self.app.knobs} for op in self.app.ops } - self.baseline_tensor = self.output_f({}) + self.baseline_tensor = None @property def name(self) -> str: @@ -218,7 +218,7 @@ class QoSModelP1(IQoSModel): def _init(self): dt = self.delta_tensors - btensor = self.baseline_tensor + btensor = self.baseline_tensor = self.output_f({}) if self.storage and self.storage.is_file(): for op, knob, delta_tensor in self._load(self.storage): dt[op][knob] = delta_tensor