Skip to content

Fixing bugs when training RHVAEs on GPU

Latest
Compare
Choose a tag to compare
@mrazomej mrazomej released this 10 May 23:29
· 6 commits to main since this release

This release fixes some bugs on the CUDA extension to be able to train RHVAEs on CUDA GPUs

Full Changelog: v0.0.2...v0.0.3