Skip to content

ROI pool层的GPU实现单测不过的问题 #10627

@dyning

Description

@dyning

将ROI pooling的单测进行修改如下,发现CPU预测可以通过,但是GPU预测有问题。
38 def init_test_case(self):
39 self.batch_size = 3
40 self.channels = 3
41 self.height = 100
42 self.width = 100
43
44 # n, c, h, w
45 self.x_dim = (self.batch_size, self.channels, self.height, self.width)
46
47 self.spatial_scale = 1.0 / 100.0
48 self.pooled_height = 3
49 self.pooled_width = 3
50
51 self.x = np.random.random(self.x_dim).astype('float32')

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions