Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ class PointCloudEncoder(nn.Module):
|
|
| 63 |
])
|
| 64 |
self.lin_global = nn.Linear(64, 128)
|
| 65 |
|
| 66 |
-
|
| 67 |
x = self.transform_net(x)
|
| 68 |
|
| 69 |
for i, conv in enumerate(self.convs):
|
|
|
|
| 63 |
])
|
| 64 |
self.lin_global = nn.Linear(64, 128)
|
| 65 |
|
| 66 |
+
def forward(self, x):
|
| 67 |
x = self.transform_net(x)
|
| 68 |
|
| 69 |
for i, conv in enumerate(self.convs):
|