童话镇里的星河

摘要： 和上一篇 SAC In JAX 一样，我们用 JAX 实现 VAE，配置一样，只需要安装符合版本的 torchvision 即可，实现中提供了 tensorboard 记录以及最后的可视化展示，测试集即为最经典的 MNIST，代码如下： import jax import flax import m 阅读全文