class IntroVAE[source]

IntroVAE(device, args) :: AbsModel

Modification of the IntroVAE-Paper for 3-Dimensional tasks in MR-Imaging based on: https://arxiv.org/abs/1807.06358 modified from: https://github.com/woxuankai/IntroVAE-Pytorch

args.model_type = "introvae"
args.dim = 3
args = compat_args(args)
test_one_batch(args)
Model-Type: introvae
args.dim = 2
args = compat_args(args)
test_one_batch(args)
Model-Type: introvae