Spaces:
Sleeping
Sleeping
Update openshape/__init__.py
Browse files- openshape/__init__.py +1 -1
openshape/__init__.py
CHANGED
@@ -35,7 +35,7 @@ def B32(s):
|
|
35 |
def S14(s):
|
36 |
model = Projected(
|
37 |
PointPatchTransformer(256, 6, 4, 1024, 96, 64, 0.4, 256, 6),
|
38 |
-
nn.Linear(256,
|
39 |
)
|
40 |
model.load_state_dict(module(['state_dict'], 'module'))
|
41 |
return model
|
|
|
35 |
def S14(s):
|
36 |
model = Projected(
|
37 |
PointPatchTransformer(256, 6, 4, 1024, 96, 64, 0.4, 256, 6),
|
38 |
+
nn.Linear(256, 1280)
|
39 |
)
|
40 |
model.load_state_dict(module(['state_dict'], 'module'))
|
41 |
return model
|