update
Browse files- networks/growth_mlp.py +0 -1
networks/growth_mlp.py
CHANGED
|
@@ -30,7 +30,6 @@ class GrowthNet(SimpleDenseNet):
|
|
| 30 |
if t.dim() < 2:
|
| 31 |
t = t[:, None]
|
| 32 |
x = torch.cat([t, x], dim=-1)
|
| 33 |
-
x = self.model(x)
|
| 34 |
x = self.softplus(self.model(x))
|
| 35 |
if self.negative:
|
| 36 |
x = -x
|
|
|
|
| 30 |
if t.dim() < 2:
|
| 31 |
t = t[:, None]
|
| 32 |
x = torch.cat([t, x], dim=-1)
|
|
|
|
| 33 |
x = self.softplus(self.model(x))
|
| 34 |
if self.negative:
|
| 35 |
x = -x
|