Lightning-Universe / lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.
https://lightning-bolts.readthedocs.io
Apache License 2.0
1.66k stars 321 forks source link

dcgan_module #971

Open yitao416 opened 1 year ago

yitao416 commented 1 year ago

🐛 Bug

In the description of dcgan_module.py, it says we can use it on cifar10 as

# cifar10
python dcgan_module.py --gpus 1 --dataset cifar10 --image_channels 3

But in the argument parser, it only has two choices:

parser.add_argument("--dataset", default="mnist", type=str, choices=["lsun", "mnist"])

Is this a typo? Or the CIFAR10 addition is down the road.

ishandutta0098 commented 1 year ago

@Borda shall I update the dcgan description or add the support for CIFAR10 to the module?

Borda commented 1 year ago

update the dcgan description or add the support for CIFAR10 to the module?

that would be great :purple_heart: