Compare commits
2 Commits
4c4297f2bd
...
bd1738dde8
| Author | SHA1 | Date | |
|---|---|---|---|
| bd1738dde8 | |||
| 754c0ee186 |
5
.gitignore
vendored
5
.gitignore
vendored
@ -1,2 +1,5 @@
|
||||
data/*
|
||||
output/*
|
||||
output/*
|
||||
*.pth
|
||||
__pycache__/*
|
||||
*.pyc
|
||||
|
||||
38
README.md
Normal file
38
README.md
Normal file
@ -0,0 +1,38 @@
|
||||
# DDPM
|
||||
|
||||
A Simple implementation of DDPM model in PyTorch.
|
||||
|
||||

|
||||

|
||||
|
||||
## Traning
|
||||
|
||||
Before training, please set up the config.ini file:
|
||||
|
||||
```ini
|
||||
[unet]
|
||||
batch_size = 256
|
||||
time_emb_dim = 128
|
||||
device = cuda
|
||||
epoch_num = 500
|
||||
learning_rate = 1e-4
|
||||
|
||||
[ddpm]
|
||||
iteration = 500
|
||||
```
|
||||
|
||||
To start training, run:
|
||||
|
||||
```
|
||||
$ python train.py
|
||||
```
|
||||
|
||||
## Sampling
|
||||
|
||||
To generate 16 pictures, run the following command:
|
||||
|
||||
The pictures will be output to the `./output` directory.
|
||||
|
||||
```
|
||||
$ python sample 16
|
||||
```
|
||||
BIN
images/Screenshot_20230314_225320.png
Normal file
BIN
images/Screenshot_20230314_225320.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 26 KiB |
BIN
images/ezgif.com-gif-maker.gif
Normal file
BIN
images/ezgif.com-gif-maker.gif
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 251 KiB |
Loading…
Reference in New Issue
Block a user