args.model_type = "bigan"
args.dim = 3
args = compat_args(args)
test_one_batch(args)
args.model_type = "bigan"
args.dim = 2
args = compat_args(args)
test_one_batch(args)
args.model_type = "bigan"
args.dim = 3
args = compat_args(args)
test_one_batch(args)
args.model_type = "bigan"
args.dim = 2
args = compat_args(args)
test_one_batch(args)