fix dimensionality of the test

jeremyfix · Jan 23, 2024 · 51b0984 · 51b0984
1 parent bc02691
commit 51b0984
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Webpage/03-pytorch-gan.md b/Webpage/03-pytorch-gan.md
@@ -104,7 +104,7 @@ def test_discriminator():
  critic = Discriminator((1, 28, 28), dropout=0.5, base_c=32, dnoise=0.1, num_classes=2)
  X = torch.randn(64, 1, 28, 28)
  out = critic(X)
- assert(out.shape == torch.Size([64]))
+ assert(out.shape == torch.Size([64, 2]))
 ```