m_step for LinearRegressionHMMEmissions and DiagonalGaussianHMMEmissions does not check whether parameters are trainable

probml / dynamax

State Space Models library in JAX

https://probml.github.io/dynamax/

MIT License

634 stars 70 forks source link

Open ghuckins opened 5 months ago

ghuckins commented 5 months ago

Might also be true for other classes of emissions; these are just the ones I noticed.