MCPcopy Create free account

hub / github.com/Royalvice/DocDiff / types & classes

Types & classes20 in github.com/Royalvice/DocDiff

↓ 5 callersClassSwish
### Swish activation function $$x \cdot \sigma(x)$$
model/DocDiff.py:9
↓ 4 callersClassResidualBlock
### Residual block A residual block has two convolution layers with group normalization. Each resolution is processed with two residua
model/DocDiff.py:61
↓ 4 callersClassSchedule
schedule/schedule.py:5
↓ 3 callersClassGaussianDiffusion
schedule/diffusionSample.py:24
↓ 2 callersClassDocData
data/docdata.py:24
↓ 2 callersClassDocDiff
model/DocDiff.py:323
↓ 2 callersClassLaplacian
src/sobel.py:26
↓ 2 callersClassTrainer
src/trainer.py:34
↓ 2 callersClassUNet
## U-Net
model/DocDiff.py:209
↓ 2 callersClassUpBlock
### 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 callersClassConfig
src/config.py:5
↓ 1 callersClassDPM_Solver
schedule/dpm_solver_pytorch.py:345
↓ 1 callersClassDownBlock
### 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 callersClassDownsample
### Scale down the feature map by $\frac{1}{2} \times$
model/DocDiff.py:193
↓ 1 callersClassEMA
model/DocDiff.py:340
↓ 1 callersClassMiddleBlock
### Middle block It combines a `ResidualBlock`, `AttentionBlock`, followed by another `ResidualBlock`. This block is applied at the lo
model/DocDiff.py:151
↓ 1 callersClassNoiseScheduleVP
schedule/dpm_solver_pytorch.py:6
↓ 1 callersClassTimeEmbedding
### Embeddings for $t$
model/DocDiff.py:19
↓ 1 callersClassUpsample
### Scale up the feature map by $2 \times$
model/DocDiff.py:177
ClassSobel
src/sobel.py:5