search

flatironinstitute / catvae

Categorical Variational Autoencoders
BSD 3-Clause "New" or "Revised" License
22 stars 3 forks source link

INF: Enable both GPU and CPU computation #46

Closed mortonjt closed 3 years ago

mortonjt commented 3 years ago

This is especially important since most people running this won't have access to GPUs