When we compiled the inference step with TensorRT we saw around ... wh], dim=-1) * self.scale_wh_delta cxycxy = torch.cat([xy, xy], dim=-1).
確定! 回上一頁