Bug fix in Annealing neuron

lava-nc · Mar 1, 2024 · ce2c270 · ce2c270
1 parent 10e199c
commit ce2c270
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/src/lava/lib/optimization/solvers/generic/annealing/process.py b/src/lava/lib/optimization/solvers/generic/annealing/process.py
@@ -77,6 +77,12 @@ def __init__(
         self.delta_temperature_out = OutPort(shape=shape)
 
         self.temperature = Var(shape=shape, init=np.int_(max_temperature))
+        self.steps_per_temperature = Var(
+            shape=shape, init=np.int_(steps_per_temperature)
+        )
+        self.temperature_counter = Var(
+            shape=shape, init=np.int_(steps_per_temperature)
+        )
 
     @property
     def shape(self) -> ty.Tuple[int, ...]:

diff --git a/src/lava/lib/optimization/solvers/generic/monitoring_processes/solution_readout/models.py b/src/lava/lib/optimization/solvers/generic/monitoring_processes/solution_readout/models.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2022 Intel Corporation
+# Copyright (C) 2022-2024 Intel Corporation
 # SPDX-License-Identifier: BSD-3-Clause
 # See: https://spdx.org/licenses/
 import numpy as np

diff --git a/src/lava/lib/optimization/solvers/generic/nebm/models.py b/src/lava/lib/optimization/solvers/generic/nebm/models.py
@@ -1,3 +1,7 @@
+# Copyright (C) 2022-2024 Intel Corporation
+# SPDX-License-Identifier: BSD-3-Clause
+# See: https://spdx.org/licenses/
+
 import numpy as np
 
 from lava.lib.optimization.solvers.generic.nebm.process import NEBM