Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Royalvice/DocDiff
/ types & classes
Types & classes
20 in github.com/Royalvice/DocDiff
⨍
Functions
101
◇
Types & classes
20
↓ 5 callers
Class
Swish
### Swish activation function $$x \cdot \sigma(x)$$
model/DocDiff.py:9
↓ 4 callers
Class
ResidualBlock
### Residual block A residual block has two convolution layers with group normalization. Each resolution is processed with two residua
model/DocDiff.py:61
↓ 4 callers
Class
Schedule
schedule/schedule.py:5
↓ 3 callers
Class
GaussianDiffusion
schedule/diffusionSample.py:24
↓ 2 callers
Class
DocData
data/docdata.py:24
↓ 2 callers
Class
DocDiff
model/DocDiff.py:323
↓ 2 callers
Class
Laplacian
src/sobel.py:26
↓ 2 callers
Class
Trainer
src/trainer.py:34
↓ 2 callers
Class
UNet
## U-Net
model/DocDiff.py:209
↓ 2 callers
Class
UpBlock
### Up block This combines `ResidualBlock` and `AttentionBlock`. These are used in the second half of U-Net at each resolution.
model/DocDiff.py:134
↓ 1 callers
Class
Config
src/config.py:5
↓ 1 callers
Class
DPM_Solver
schedule/dpm_solver_pytorch.py:345
↓ 1 callers
Class
DownBlock
### Down block This combines `ResidualBlock` and `AttentionBlock`. These are used in the first half of U-Net at each resolution.
model/DocDiff.py:119
↓ 1 callers
Class
Downsample
### Scale down the feature map by $\frac{1}{2} \times$
model/DocDiff.py:193
↓ 1 callers
Class
EMA
model/DocDiff.py:340
↓ 1 callers
Class
MiddleBlock
### Middle block It combines a `ResidualBlock`, `AttentionBlock`, followed by another `ResidualBlock`. This block is applied at the lo
model/DocDiff.py:151
↓ 1 callers
Class
NoiseScheduleVP
schedule/dpm_solver_pytorch.py:6
↓ 1 callers
Class
TimeEmbedding
### Embeddings for $t$
model/DocDiff.py:19
↓ 1 callers
Class
Upsample
### Scale up the feature map by $2 \times$
model/DocDiff.py:177
Class
Sobel
src/sobel.py:5