Skip to content

Commit c8c18d1

Browse files
yang-songyang-song
committed
first release
1 parent 235991b commit c8c18d1

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

66 files changed

+787
-1542
lines changed

‎README.md

Lines changed: 62 additions & 9 deletions

‎assets/bedroom.jpeg

75.8 KB

‎assets/celebahq_256.jpg

71.7 KB

‎assets/church.jpeg

121 KB

‎assets/ffhq_1024.jpeg

139 KB

‎assets/ffhq_256.jpg

70.1 KB

‎configs/default_celeba_configs.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import ml_collections
2+
import torch
23

34

45
def get_default_configs():
@@ -16,7 +17,6 @@ def get_default_configs():
1617
training.snapshot_sampling = True
1718
training.likelihood_weighting = False
1819
training.continuous = True
19-
training.n_jitted_steps = 5
2020
training.reduce_mean = False
2121

2222
# sampling
@@ -67,5 +67,6 @@ def get_default_configs():
6767
optim.grad_clip = 1.
6868

6969
config.seed = 42
70+
config.device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
7071

7172
return config

‎configs/default_cifar10_configs.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import ml_collections
2+
import torch
23

34

45
def get_default_configs():
@@ -16,7 +17,6 @@ def get_default_configs():
1617
training.snapshot_sampling = True
1718
training.likelihood_weighting = False
1819
training.continuous = True
19-
training.n_jitted_steps = 5
2020
training.reduce_mean = False
2121

2222
# sampling
@@ -67,5 +67,6 @@ def get_default_configs():
6767
optim.grad_clip = 1.
6868

6969
config.seed = 42
70+
config.device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
7071

7172
return config

‎configs/default_lsun_configs.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import ml_collections
2+
import torch
23

34

45
def get_default_configs():
@@ -16,7 +17,6 @@ def get_default_configs():
1617
training.snapshot_sampling = True
1718
training.likelihood_weighting = False
1819
training.continuous = True
19-
training.n_jitted_steps = 5
2020
training.reduce_mean = False
2121

2222
# sampling
@@ -67,5 +67,6 @@ def get_default_configs():
6767
optim.grad_clip = 1.
6868

6969
config.seed = 42
70+
config.device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
7071

7172
return config

‎configs/ddpm/cifar10_continuous_subvp.py renamed to ‎configs/subvp/cifar10_ddpm_continuous.py

File renamed without changes.

0 commit comments

Comments
 (0)