masa-su / pixyz

A library for developing deep generative models in a more concise, intuitive and extendable way

https://pixyz.io

MIT License

487 stars 41 forks source link

Develop/v0.3.3 #175

Closed masa-su closed 2 years ago

masa-su commented 2 years ago

new features

add 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.
add the mixture of expert (MoE)

bug fix

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