Skip to content

v0.3.3

Latest
Compare
Choose a tag to compare
@masa-su masa-su released this 14 Dec 14:26
· 2 commits to main since this release
a9baf06

New features

  • Added set_cache_maxsize, a setter of maxsize for lru_cache, and cache_maxsize, a getter of maxsize for lru_cache. By increasing this maxsize, the size that can store the results of get_params (≒ forward of the Distribution class) becomes larger. This will speed up most VAE implementations!
  • Added the mixture of experts (MoE)

Bug fix

  • Fixed get_params for replace_var
  • Fixed maxsize of lru_cache for get_params
  • Fixed a major bug in PoE.