We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
将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')
The text was updated successfully, but these errors were encountered:
赞!
Sorry, something went wrong.
baiyfbupt
Successfully merging a pull request may close this issue.
将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')
The text was updated successfully, but these errors were encountered: