在resnet101的最后一个block的feature size和膨胀卷积问题

[niujiaopeng]

非常好的代码
我看了作者的tensorflow代码，问一个问题，是不是您这也是在标准的resnet101上修改的
普通的resnet的block4的feature size，也就是stage2的feature size应该是block3(rpn feature size)的二分之一，比如说提取rpn feature的size是50x75，那么最后进入那个head的应该是25x38，但是这里貌似最后的size还是50x75，block4和block3是一样的size，并且还用了2次膨胀卷积，这里的预训练参数是您完全按照这个网络结构重新在imagenet 上重新训练的呢，还是直接用标准的resnet的预训练参数？

[my-hello-world]

@niujiaopeng @TreB1eN 你好，请问你跑通了吗？我出现了问题

psroialign_cuda.cpp: In function ‘int psroialign_backward_cuda(at::Tensor, at::Tensor, at::Tensor, at::Tensor, float, int, int)’:
psroialign_cuda.cpp:25:80: error: ‘AT_CHECK’ was not declared in this scope
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
                                                                                ^
psroialign_cuda.cpp:29:3: note: in expansion of macro ‘CHECK_CUDA’
   CHECK_CUDA(x);       \
   ^
psroialign_cuda.cpp:66:5: note: in expansion of macro ‘CHECK_INPUT’
     CHECK_INPUT(top_diff);
     ^
error: command 'gcc' failed with exit status 1

请问怎么解决呢？

[TreB1eN]

应该是pytorch的版本问题，你用的是老版本？

[my-hello-world]

@TreB1eN 没哎，用的0.4.0

[TreB1eN]

升级一下呢，我记得我遇到过这个问题，是版本的问题，你也可以google一下

[mmaaz60]

hi, which CUDA and CUDNN version work fine for you?