Harshit Agarwal
commited on
Commit
·
ee90412
1
Parent(s):
3896b0b
slides added
Browse files- model/model.py +1 -1
- model/precomputes.py +1 -1
- slides/DDPM - Basics (1).pdf +0 -0
model/model.py
CHANGED
@@ -14,7 +14,7 @@ class Block(nn.Module):
|
|
14 |
super().__init__()
|
15 |
self.time_mlp = nn.Linear(time_emb_dim, out_ch)
|
16 |
if up:
|
17 |
-
## up channel -
|
18 |
self.conv1 = nn.Conv2d(2*in_ch, out_ch, 3, padding=1)
|
19 |
self.transform = nn.ConvTranspose2d(out_ch, out_ch, 4, 2, 1)
|
20 |
else:
|
|
|
14 |
super().__init__()
|
15 |
self.time_mlp = nn.Linear(time_emb_dim, out_ch)
|
16 |
if up:
|
17 |
+
## up channel - gobig big big bigg from smol smol smol with 3x3 kernel
|
18 |
self.conv1 = nn.Conv2d(2*in_ch, out_ch, 3, padding=1)
|
19 |
self.transform = nn.ConvTranspose2d(out_ch, out_ch, 4, 2, 1)
|
20 |
else:
|
model/precomputes.py
CHANGED
@@ -3,7 +3,7 @@ from torch.nn import functional as F
|
|
3 |
|
4 |
T = 300 ## according to the paper
|
5 |
|
6 |
-
### SOO MMANNYY PRECOMPUTEDD VALUESS TO
|
7 |
betas = torch.linspace(1e-4, 0.02, T)
|
8 |
alphas = 1. - betas
|
9 |
alphas_cumulative_products = torch.cumprod(alphas, axis=0)
|
|
|
3 |
|
4 |
T = 300 ## according to the paper
|
5 |
|
6 |
+
### SOO MMANNYY PRECOMPUTEDD VALUESS TO TRACKKKKSS
|
7 |
betas = torch.linspace(1e-4, 0.02, T)
|
8 |
alphas = 1. - betas
|
9 |
alphas_cumulative_products = torch.cumprod(alphas, axis=0)
|
slides/DDPM - Basics (1).pdf
ADDED
The diff for this file is too large to render.
See raw diff
|
|