Hello sir,
I tried to use FPN in ResNet50 model but i couldn't train my model yet. As i know if i want to use FPN i need to remove these two lines
"" self.avgpool = nn.AdaptiveAvgPool2d((1, 1))""
""self.fc = nn.Linear(512 * block.expansion, num_classes)"".
but if i remove them, i can't train the model. so could you please help me to solve this problem or suggest me any blogs!?
thanks!
RY-97
commented
2 years ago
Hello sir, I tried to use FPN in ResNet50 model but i couldn't train my model yet. As i know if i want to use FPN i need to remove these two lines "" self.avgpool = nn.AdaptiveAvgPool2d((1, 1))"" ""self.fc = nn.Linear(512 * block.expansion, num_classes)"".
but if i remove them, i can't train the model. so could you please help me to solve this problem or suggest me any blogs!? thanks!