learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-10-30 02:08:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

__call__ -> forward

Browse Source
This commit is contained in:
Varuna Jayasiri
2021-08-19 15:47:25 +05:30
parent eaa248c9e6
commit 02309fa4cc
5 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
labml_nn/gan/dcgan/__init__.py
View File

@ -53,7 +53,7 @@ class Generator(Module):
self.apply(_weights_init)
def __call__(self, x):
def forward(self, x):
# Change from shape `[batch_size, 100]` to `[batch_size, 100, 1, 1]`
x = x.unsqueeze(-1).unsqueeze(-1)
x = self.layers(x)