Skip to content

Commit a8ddc7d

Browse files
authored
Merge pull request #223 from tatsuookubo/master
Very minor typos in the type hint and text description
2 parents 9a42ac2 + 3a759f4 commit a8ddc7d

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

labml_nn/diffusion/ddpm/unet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ class UNet(Module):
312312

313313
def __init__(self, image_channels: int = 3, n_channels: int = 64,
314314
ch_mults: Union[Tuple[int, ...], List[int]] = (1, 2, 2, 4),
315-
is_attn: Union[Tuple[bool, ...], List[int]] = (False, False, True, True),
315+
is_attn: Union[Tuple[bool, ...], List[bool]] = (False, False, True, True),
316316
n_blocks: int = 2):
317317
"""
318318
* `image_channels` is the number of channels in the image. $3$ for RGB.

labml_nn/normalization/group_norm/readme.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,6 @@ This is based on the observation that classical features such as
1515
The paper proposes dividing feature channels into groups and then separately normalizing
1616
all channels within each group.
1717

18-
Here's a [CIFAR 10 classification model](https://nn.labml.ai/normalization/group_norm/experiment.html) that uses instance normalization.
18+
Here's a [CIFAR 10 classification model](https://nn.labml.ai/normalization/group_norm/experiment.html) that uses group normalization.
1919

2020
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/labmlai/annotated_deep_learning_paper_implementations/blob/master/labml_nn/normalization/group_norm/experiment.ipynb)

0 commit comments

Comments
 (0)