How can I run the model using the ResNeXt features?
I tried to run the model with FPN.CONV_BODY: FPN.add_fpn_ResNet152_conv5_body
and with the pretrained X-152-32x8d-IN5k.pkl file and got an assertion error:
"Workspace blob res2_0_branch2a_w with shape (64, 64, 1, 1) does not match weights file shape (256, 64, 1, 1)"
So I believe there should be another conv body for the ResNeXt case? couldn't find it in FPN so I'm not sure what value I should set it to.
How can I run the model using the ResNeXt features? I tried to run the model with FPN.CONV_BODY: FPN.add_fpn_ResNet152_conv5_body and with the pretrained X-152-32x8d-IN5k.pkl file and got an assertion error: "Workspace blob res2_0_branch2a_w with shape (64, 64, 1, 1) does not match weights file shape (256, 64, 1, 1)"
So I believe there should be another conv body for the ResNeXt case? couldn't find it in FPN so I'm not sure what value I should set it to.
Thanks a lot!