winfred2027 commited on
Commit
428cfb2
·
verified ·
1 Parent(s): 7134167

Update openshape/__init__.py

Browse files
Files changed (1) hide show
  1. 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, cfg.model.out_channel)
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