Commit Graph

20 Commits

Author SHA1 Message Date
9852bf3530
feat: classifier guidence diffusion model 2023-03-22 16:15:09 +08:00
687a994bea
docs: update readme 2023-03-14 23:33:54 +08:00
bd1738dde8
docs: add readme 2023-03-14 23:28:08 +08:00
754c0ee186
docs: update .gitignore 2023-03-14 23:21:22 +08:00
4c4297f2bd
style: use configparser to manage config file 2023-03-14 23:20:32 +08:00
81779cf5e2
docs: delete useless comment 2023-03-14 23:19:21 +08:00
667346c561
feat: complete sample.py 2023-03-14 22:58:07 +08:00
a330666a17
fix: set n in sample() default value to self.batch_size 2023-03-14 22:57:44 +08:00
37adcc7e97
feat: check whether generating seq pic 2023-03-14 22:46:53 +08:00
d3eff8e425
docs: update .gitignore 2023-03-14 18:47:17 +08:00
38fa577706
fix: decrease batch size, iteration, epoch and lr 2023-03-14 18:45:04 +08:00
3530d91aaf
fix: add dropout layer into unet model 2023-03-14 18:18:13 +08:00
79e8f37f5b
fix: dont add noise when t=0 && save iteration figure in DDPM sample 2023-03-14 18:16:56 +08:00
25e6a5ff62
feat: add sample.py 2023-03-14 02:16:06 +08:00
7a0b5a2a8a
fix: remove nn.Tanh() on the last layer 2023-03-14 02:14:19 +08:00
fab0bc4af1
fix: fix typo & move tensor to cuda device 2023-03-14 02:12:58 +08:00
dcfeac845e
feat: complete ddpm & training process 2023-03-14 00:08:54 +08:00
816f6d1f56
feat: DDPM add noise & get time sequence 2023-03-13 22:33:04 +08:00
ed018d3cd9
docs: add .gitignore 2023-03-13 22:30:05 +08:00
9373320e4a
feat: Unet module 2023-03-13 21:29:16 +08:00