Varitional Autoencoder

Variational Autoencoder for Resting-State Functional MRI

2D, 3D, and graph convolutional VAEs built in Pytorch for use with rsfMRI data