-
Notifications
You must be signed in to change notification settings - Fork 5.9k
Closed
Labels
Description
将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')