class DCGAN[source]

DCGAN(device, args) :: AbsModel

Modification of the DCGAN-Paper https://arxiv.org/pdf/1511.06434.pdf for 3-Dimensional tasks in MR-Imaging oriented on: https://pytorch.org/tutorials/beginner/dcgan_faces_tutorial.html

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