diffkalman 0.1.0__tar.gz → 0.1.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffkalman
3
- Version: 0.1.0
3
+ Version: 0.1.1
4
4
  Summary: A diffrentiable kalman filter library for auto-tuning kalman filters.
5
5
  Author-email: hades <nischalbhattaraipi@gmail.com>
6
6
  License-File: LICENSE
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "diffkalman"
3
- version = "0.1.0"
3
+ version = "0.1.1"
4
4
  description = "A diffrentiable kalman filter library for auto-tuning kalman filters."
5
5
  readme = "README.md"
6
6
  authors = [
@@ -103,12 +103,15 @@ def em_updates(
103
103
  # Update the parameters
104
104
  (-complete_log_likelihood).backward()
105
105
  optimizer.step()
106
- lr_scheduler.step()
106
+
107
107
 
108
108
  # Print the log likelihood
109
109
  print(
110
110
  f"Epoch {e + 1}/{num_epochs} Cycle {c + 1}/{num_cycles} Log Likelihood: {marginal_likelihood.item()}"
111
111
  )
112
+
113
+ # Update the learning rate
114
+ lr_scheduler.step()
112
115
 
113
116
  return {
114
117
  "likelihoods": likelihoods,
File without changes
File without changes
File without changes
File without changes
File without changes