MCPcopy Create free account
hub / github.com/FreeformRobotics/OTS / forward

Method forward

models/ots.py:59–67  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

57 weight_init(layer)
58
59 def forward(self, x):
60 x = self.attn1(x)
61 x = self.attn2(x)
62 x = self.depth_conv(x)
63 x = self.point_conv(x)
64 x = self.norm(x)
65 x = self.relu(x)
66 x = x.view(x.size(0), -1)
67 return x
68
69
70class Classifier(nn.Module):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected