Skip to content

khoatranrb/ts_v2.0

Repository files navigation

Unoffical implement DDPNOpt

Classify task

  • To train with Adam/RMSprop, use _mnist.py.
  • To train with DDPNOpt, use _mnist_ddp.py.

Denoise task

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages