ROI pool层的GPU实现单测不过的问题
Created by: 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')