Compare commits
No commits in common. "bd1738dde833648c792f38bac02fd7551e6938bc" and "4c4297f2bd9b3cbf2f5da63b2827db5368f1e3ef" have entirely different histories.
bd1738dde8
...
4c4297f2bd
5
.gitignore
vendored
5
.gitignore
vendored
@ -1,5 +1,2 @@
|
|||||||
data/*
|
data/*
|
||||||
output/*
|
output/*
|
||||||
*.pth
|
|
||||||
__pycache__/*
|
|
||||||
*.pyc
|
|
||||||
38
README.md
38
README.md
@ -1,38 +0,0 @@
|
|||||||
# 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
|
|
||||||
```
|
|
||||||
Binary file not shown.
|
Before Width: | Height: | Size: 26 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 251 KiB |
Loading…
Reference in New Issue
Block a user