在resnet101的最后一个block的feature size和膨胀卷积问题
niujiaopeng opened this issue · 5 comments
niujiaopeng commented
非常好的代码
我看了作者的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 commented
@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 commented
应该是pytorch的版本问题,你用的是老版本?
my-hello-world commented
@TreB1eN 没哎,用的0.4.0
TreB1eN commented
升级一下呢,我记得我遇到过这个问题,是版本的问题,你也可以google一下
mmaaz60 commented
hi, which CUDA and CUDNN version work fine for you?