MCPcopy Create free account
hub / github.com/MotrixLab/AiOS / init_ref_points

Method init_ref_points

models/aios/transformer.py:2293–2301  ·  view source on GitHub ↗
(self, use_num_queries)

Source from the content-addressed store, hash-verified

2291 return valid_ratio
2292
2293 def init_ref_points(self, use_num_queries):
2294 self.refpoint_embed = nn.Embedding(use_num_queries, 4)
2295
2296 if self.random_refpoints_xy:
2297 # import pdb; pdb.set_trace()
2298 self.refpoint_embed.weight.data[:, :2].uniform_(0, 1)
2299 self.refpoint_embed.weight.data[:, :2] = inverse_sigmoid(
2300 self.refpoint_embed.weight.data[:, :2])
2301 self.refpoint_embed.weight.data[:, :2].requires_grad = False
2302
2303 # srcs: features; refpoint_embed:
2304 def forward(self,

Callers 1

__init__Method · 0.95

Calls 1

inverse_sigmoidFunction · 0.90

Tested by

no test coverage detected