- To train with Adam/RMSprop, use
_mnist.py
. - To train with DDPNOpt, use
_mnist_ddp.py
.
- Download full data at GGDrive.
- To train with Adam/RMSprop, use
_denoise.py
. - To train with DDPNOpt, use
_denoise_ddp.py
.
- Download model at saved.
- Use file
_infer_ddp.py
.