forked from mrq/DL-Art-School
.
This commit is contained in:
parent
dee2b72786
commit
91be38cba3
|
@ -27,7 +27,7 @@ class ConditioningEncoder(nn.Module):
|
|||
|
||||
def forward(self, x):
|
||||
h = checkpoint(self.init, x)
|
||||
h = self.attn(h
|
||||
h = self.attn(h)
|
||||
return h.mean(dim=2)
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user